diff --git a/src/View/Reservoir/Edit/Window.py b/src/View/Reservoir/Edit/Window.py
index ed8b2cd38440cdaffaca03caf91414db36d5cca9..84922a19149dcac12da8167419ca5358c79830df 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()