diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py
index 839205ca2..dc5e6db5e 100644
--- a/src/backend/langflow/graph/graph/base.py
+++ b/src/backend/langflow/graph/graph/base.py
@@ -1,4 +1,5 @@
 from typing import Dict, Generator, List, Type, Union
+from collections import defaultdict, deque
 
 from langchain.chains.base import Chain
 from loguru import logger