diff --git a/arcaflow_plugin_wait/arcaflow_plugin_wait.py b/arcaflow_plugin_wait/arcaflow_plugin_wait.py
index 906a1aa..fae2d7a 100644
--- a/arcaflow_plugin_wait/arcaflow_plugin_wait.py
+++ b/arcaflow_plugin_wait/arcaflow_plugin_wait.py
@@ -60,7 +60,11 @@ def cancel_step(self, _input: predefined_schemas.cancelInput):
         id="wait",
         name="Wait",
         description="Waits for the given amount of time",
-        outputs={"success": SuccessOutput, "error": ErrorOutput, "cancelled_early": ErrorOutput},
+        outputs={
+            "success": SuccessOutput,
+            "error": ErrorOutput,
+            "cancelled_early": ErrorOutput
+        },
         signal_handler_method_names=["cancel_step"],
         signal_emitters=[],
         step_object_constructor=lambda: WaitStep(),