diff --git a/debye_bec/bec_widgets/widgets/scheduler/item_dialog.py b/debye_bec/bec_widgets/widgets/scheduler/item_dialog.py
index b5a0315..e79f92e 100644
--- a/debye_bec/bec_widgets/widgets/scheduler/item_dialog.py
+++ b/debye_bec/bec_widgets/widgets/scheduler/item_dialog.py
@@ -252,7 +252,12 @@ class ScheduleItemDialog(QDialog):
         return {
             "command": f"dev.{device_name}.move({value!r}, relative={relative!r})",
             "kind": "move",
-            "form_state": {"device_name": device_name, "value": value, "units": units, "relative": relative},
+            "form_state": {
+                "device_name": device_name,
+                "value": value,
+                "units": units,
+                "relative": relative,
+            },
         }
 
     def _prefill_move_page(self, state: dict):
@@ -554,7 +559,7 @@ class ScheduleItemDialog(QDialog):
             raise ValueError("Select at least one amplifier for auto gain.")
         cmd = (
             f"auto_gain.start(element={self.edge_element}, edge={self.edge_edge}, "
-            f"{amplifiers}, comp_ring_current=True)"
+            f"amplifier={amplifiers}, comp_ring_current=True)"
         )
         return {
             "command": cmd,