diff --git a/src/View/Reservoir/Edit/Window.py b/src/View/Reservoir/Edit/Window.py
index ed8b2cd3..84922a19 100644
--- a/src/View/Reservoir/Edit/Window.py
+++ b/src/View/Reservoir/Edit/Window.py
@@ -202,10 +202,8 @@ class EditReservoirWindow(PamhyrWindow):
 
     def _undo(self):
         self._table.undo()
-        self.plot.update()
-        self.widget_update()
+        self.update()
 
     def _redo(self):
         self._table.redo()
-        self.plot.update()
-        self.widget_update()
+        self.update()