diff --git a/src/View/Geometry/Profile/ProfileWindow.py b/src/View/Geometry/Profile/ProfileWindow.py
index c4f57867648bcb443634e4b75a11498a7bab3f4b..937b6e5f9c6f031da2487844c80fadb3d09273d3 100644
--- a/src/View/Geometry/Profile/ProfileWindow.py
+++ b/src/View/Geometry/Profile/ProfileWindow.py
@@ -134,10 +134,10 @@ class ProfileWindow(QMainWindow):
 
     def insert_row(self):
         if len(self.ui.tableView.selectedIndexes()) == 0:
-            self._tablemodel.insert_row(self._tablemodel.rowCount())
+            self._model.insert_row(self._model.rowCount())
         else:
             row = self.index_selected_row()
-            self._tablemodel.insert_row(row + 1)
+            self._model.insert_row(row + 1)
 
     def delete_row(self):
         rows = sorted(
@@ -149,7 +149,7 @@ class ProfileWindow(QMainWindow):
         )
 
         if len(rows) > 0:
-            self._tablemodel.remove_rows(rows)
+            self._model.remove_rows(rows)
 
     def sort_X_ascending(self):
         self._model.sort('x', order=Qt.AscendingOrder)
@@ -209,20 +209,16 @@ class ProfileWindow(QMainWindow):
 
     def paste(self):
         row = self.index_selected_row()
-        self._tablemodel.paste(row, self._clipboard)
+        self._model.paste(row, self._clipboard)
         self.select_current_profile()
 
     def undo(self):
-        self._tablemodel.undo()
-        self.select_current_profile()
-        self.update_plot_xy()
-        self.update_plot_kpc()
+        self._model.undo()
+        self.update_plot()
 
     def redo(self):
-        self._tablemodel.redo()
-        self.select_current_profile()
-        self.update_plot_xy()
-        self.update_plot_kpc()
+        self._model.redo()
+        self.update_plot()
 
     def handleSave(self):
         if self.fileName is None or self.fileName == '':
diff --git a/src/View/Geometry/Profile/qtableview_profile.py b/src/View/Geometry/Profile/qtableview_profile.py
index 6c794434d2439690dec80a3ec8aef2ecc83c97c2..0fa62ee9e0f45e1e795718e5dc3939c7ae02baee 100644
--- a/src/View/Geometry/Profile/qtableview_profile.py
+++ b/src/View/Geometry/Profile/qtableview_profile.py
@@ -137,7 +137,7 @@ class TableEditableModel(QAbstractTableModel):
                 self._undo_stack.push(
                     SetXCommand(
                         self._profile, row,
-                        self._profile.profile(row).x,
+                        self._profile.point(row).x,
                         value
                     )
                 )
@@ -145,7 +145,7 @@ class TableEditableModel(QAbstractTableModel):
                 self._undo_stack.push(
                     SetYCommand(
                         self._profile, row,
-                        self._profile.profile(row).y,
+                        self._profile.point(row).y,
                         value
                     )
                 )
@@ -153,7 +153,7 @@ class TableEditableModel(QAbstractTableModel):
                 self._undo_stack.push(
                     SetZCommand(
                         self._profile, row,
-                        self._profile.profile(row).z,
+                        self._profile.point(row).z,
                         value
                     )
                 )
@@ -161,7 +161,7 @@ class TableEditableModel(QAbstractTableModel):
                 self._undo_stack.push(
                     SetNameCommand(
                         self._profile, row,
-                        self._profile.profile(row).name,
+                        self._profile.point(row).name,
                         value
                     )
                 )