diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py
index f57418a9a..631d58e41 100644
--- a/src/backend/langflow/interface/run.py
+++ b/src/backend/langflow/interface/run.py
@@ -3,7 +3,9 @@ from langflow.graph import Graph
 from loguru import logger
 
 
-def build_sorted_vertices(data_graph, user_id) -> Tuple[Graph, Dict]:
+def build_sorted_vertices(
+    data_graph, user_id: Optional[Union[str, UUID]] = None
+) -> Tuple[Graph, Dict]:
     """
     Build langchain object from data_graph.
     """