diff --git a/src/backend/base/langflow/base/tools/flow_tool.py b/src/backend/base/langflow/base/tools/flow_tool.py
index dc592987d9a..9884ab52b57 100644
--- a/src/backend/base/langflow/base/tools/flow_tool.py
+++ b/src/backend/base/langflow/base/tools/flow_tool.py
@@ -6,6 +6,8 @@
 from loguru import logger
 
 from langflow.base.flow_processing.utils import build_data_from_result_data, format_flow_output_data
+from langflow.graph.graph.base import Graph  # cannot be a part of TYPE_CHECKING   # noqa: TCH001
+from langflow.graph.vertex.base import Vertex  # cannot be a part of TYPE_CHECKING  # noqa: TCH001
 from langflow.helpers.flow import build_schema_from_inputs, get_arg_names, get_flow_inputs, run_flow
 from langflow.utils.async_helpers import run_until_complete
 
@@ -13,9 +15,6 @@
     from langchain_core.runnables import RunnableConfig
     from pydantic.v1 import BaseModel
 
-    from langflow.graph.graph.base import Graph
-    from langflow.graph.vertex.base import Vertex
-
 
 class FlowTool(BaseTool):
     name: str