diff --git a/modules/shared_state.py b/modules/shared_state.py
index b9434c637..0d03ebb17 100644
--- a/modules/shared_state.py
+++ b/modules/shared_state.py
@@ -154,7 +154,7 @@ class State:
         else:
             self.results.append(results)
 
-    def get_id(self, task_id):
+    def get_id(self, task_id:str=None):
         if task_id is None or task_id == 0:
             task_id = uuid.uuid4().hex[:15]
         if not isinstance(task_id, str):