diff --git a/src/View/Geometry/Profile/Window.py b/src/View/Geometry/Profile/Window.py
index 90fe1f44b054efea646e54ad4ebe4a0116a67405..e4112f409f74684f428d62d58bb141c9d841c777 100644
--- a/src/View/Geometry/Profile/Window.py
+++ b/src/View/Geometry/Profile/Window.py
@@ -145,7 +145,8 @@ class ProfileWindow(PamhyrWindow):
                       .row()
 
     def add(self):
-        if len(self._tablemodel.selectedIndexes()) == 0:
+        table = self.find(QTableView, "tableView")
+        if len(table.selectedIndexes()) == 0:
             self._tablemodel.insert_row(self._tablemodel.rowCount())
         else:
             row = self.index_selected_row()
@@ -153,10 +154,11 @@ class ProfileWindow(PamhyrWindow):
         self.update_plot()
 
     def delete(self):
+        table = self.find(QTableView, "tableView")
         rows = sorted(
             list(
                 set(
-                    [index.row() for index in self._tablemodel.selectedIndexes()]
+                    [index.row() for index in table.selectedIndexes()]
                 )
             )
         )
@@ -184,7 +186,8 @@ class ProfileWindow(PamhyrWindow):
     def move_down(self):
         rows = list(
             set(
-                [index.row() for index in self._tablemodel.selectedIndexes()]
+                [index.row() for index in
+                 self.find(QTableView, "tableView").selectedIndexes()]
             )
         )
 
@@ -197,7 +200,8 @@ class ProfileWindow(PamhyrWindow):
     def move_up(self):
         rows = list(
             set(
-                [index.row() for index in self._tablemodel.selectedIndexes()]
+                [index.row() for index in
+                 self.find(QTableView, "tableView").selectedIndexes()]
             )
         )
 
diff --git a/src/View/Geometry/Window.py b/src/View/Geometry/Window.py
index 2fccde41ab74c8754038ca2c51cb162b9dc132aa..af27773f0051da081f4612c641c79ab748bc4c0a 100644
--- a/src/View/Geometry/Window.py
+++ b/src/View/Geometry/Window.py
@@ -36,7 +36,7 @@ from PyQt5.QtWidgets import (
     QApplication, QMainWindow, QFileDialog, QCheckBox,
     QUndoStack, QShortcut, QTableView, QHeaderView,
     QAction, QSlider, QPushButton, QVBoxLayout,
-    QLabel,
+    QLabel, QAbstractItemView,
 )
 
 from View.Tools.PamhyrWindow import PamhyrWindow