diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py
index 631d58e41..b482f8c50 100644
--- a/src/backend/langflow/interface/run.py
+++ b/src/backend/langflow/interface/run.py
@@ -1,6 +1,7 @@
-from typing import Dict, Tuple
+from typing import Dict, Tuple, Optional, Union
 from langflow.graph import Graph
 from loguru import logger
+from uuid import UUID
 
 
 def build_sorted_vertices(