diff --git a/src/View/SedimentLayers/Edit/Table.py b/src/View/SedimentLayers/Edit/Table.py
index 6bd59387046a52f975e6ae28456b8ddb19bc57fe..32790f1bebd1279a1cdd12aa2108fd297b802927 100644
--- a/src/View/SedimentLayers/Edit/Table.py
+++ b/src/View/SedimentLayers/Edit/Table.py
@@ -76,9 +76,15 @@ class TableModel(QAbstractTableModel):
                     self._sl, row, value
                 )
             )
-        if self._headers[column] == "comment":
+        if self._headers[column] == "type":
             self._undo.push(
-                SetCommentCommand(
+                SetTypeCommand(
+                    self._sl, row, value
+                )
+            )
+        if self._headers[column] == "height":
+            self._undo.push(
+                SetHeightCommand(
                     self._sl, row, value
                 )
             )
diff --git a/src/View/SedimentLayers/Edit/UndoCommand.py b/src/View/SedimentLayers/Edit/UndoCommand.py
index 9e630a6e73e93fd8b9782a37fb878b01e4256160..e69647092bb99452d377659d03fd862a2901de22 100644
--- a/src/View/SedimentLayers/Edit/UndoCommand.py
+++ b/src/View/SedimentLayers/Edit/UndoCommand.py
@@ -25,6 +25,21 @@ class SetNameCommand(QUndoCommand):
     def redo(self):
         self._sediment_layers.get(self._index).name = self._new
 
+class SetTypeCommand(QUndoCommand):
+    def __init__(self, sediment_layers, index, new_value):
+        QUndoCommand.__init__(self)
+
+        self._sediment_layers = sediment_layers
+        self._index = index
+        self._old = self._sediment_layers.get(self._index).type
+        self._new = new_value
+
+    def undo(self):
+        self._sediment_layers.get(self._index).type = self._old
+
+    def redo(self):
+        self._sediment_layers.get(self._index).type = self._new
+
 class SetHeightCommand(QUndoCommand):
     def __init__(self, sediment_layers, index, new_value):
         QUndoCommand.__init__(self)