Fix undo of widget addition

ObaraEmmanuel · Sep 20, 2023 · ba717ab · ba717ab
1 parent b5cf2da
commit ba717ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/studio/feature/design.py b/studio/feature/design.py
@@ -585,8 +585,8 @@ def add(self, obj_class: PseudoWidget.__class__, x, y, **kwargs):
             if not silent:
                 self.studio.new_action(Action(
                     # Delete silently to prevent adding the event to the undo/redo stack
-                    lambda _: self.delete(obj, True),
-                    lambda _: self.restore(obj, restore_point, obj.layout)
+                    lambda _: self.delete([obj], True),
+                    lambda _: self.restore([obj], [restore_point], [obj.layout])
                 ))
         elif obj.layout is None:
             # This only happens when adding the main layout. We dont need to add this action to the undo/redo stack