diff --git a/b_asic/GUI/drag_button.py b/b_asic/GUI/drag_button.py
index 7b4885014239d7402005899ee7d06d211021396a..85cfd7347dc4951e303ce7a846e5038d0aa9c20b 100644
--- a/b_asic/GUI/drag_button.py
+++ b/b_asic/GUI/drag_button.py
@@ -122,7 +122,7 @@ class DragButton(QPushButton):
         self._window.graphic_view.update()
         super().mouseReleaseEvent(event)
 
-    def _flip(self, event):
+    def _flip(self, event=None):
         self._flipped = not self._flipped
         for pb in self.ports:
             if isinstance(pb.port, InputPort):
diff --git a/b_asic/GUI/main_window.py b/b_asic/GUI/main_window.py
index 843a4c933751fc53f5d6389330308666e32ca80b..de1cca361311c99c21e8efee2463f26cdf9c1f5b 100644
--- a/b_asic/GUI/main_window.py
+++ b/b_asic/GUI/main_window.py
@@ -216,16 +216,14 @@ class MainWindow(QMainWindow):
 
         self.logger.info(f"Saved SFG to path: {module}.")
 
-    def save_work(self, arg):
-        print("Save: ", arg)
+    def save_work(self, event=None):
         self.sfg_widget = SelectSFGWindow(self)
         self.sfg_widget.show()
 
         # Wait for input to dialog.
         self.sfg_widget.ok.connect(self._save_work)
 
-    def load_work(self, arg):
-        print("Load: ", arg)
+    def load_work(self, event=None):
         module, accepted = QFileDialog().getOpenFileName()
         if not accepted:
             return