diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py
index 5f136b3c7..9db6260e9 100644
--- a/src/backend/langflow/graph/nodes.py
+++ b/src/backend/langflow/graph/nodes.py
@@ -8,7 +8,7 @@ class AgentNode(Node):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="agents")
 
-        self.tools: List[ToolNode] = []
+        self.tools: List[Union[ToolNode, ToolkitNode]] = []
         self.chains: List[ChainNode] = []
 
     def _set_tools_and_chains(self) -> None: