diff --git a/orbiter_translations/automic/xml_demo.py b/orbiter_translations/automic/xml_demo.py
index 1fbaac0..25fe3c0 100644
--- a/orbiter_translations/automic/xml_demo.py
+++ b/orbiter_translations/automic/xml_demo.py
@@ -302,7 +302,9 @@ def bash_command_rule(
 
 
 @task_rule(priority=2)
-def start_or_end_rule(val):
+def start_or_end_rule(
+    val: dict,
+) -> OrbiterEmptyOperator | None:
     """
     Translate input into an EmptyOperator for start and end tasks