diff --git a/src/backend/base/langflow/components/tools/astradb.py b/src/backend/base/langflow/components/tools/astradb.py
index 13f45025d..0376f2d64 100644
--- a/src/backend/base/langflow/components/tools/astradb.py
+++ b/src/backend/base/langflow/components/tools/astradb.py
@@ -3,7 +3,7 @@ from typing import Any
 
 from astrapy import Collection, DataAPIClient, Database
 from langchain.pydantic_v1 import BaseModel, Field, create_model
-from langchain_core.tools import StructuredTool
+from langchain_core.tools import StructuredTool, Tool
 
 from langflow.base.langchain_utilities.model import LCToolComponent
 from langflow.io import DictInput, IntInput, SecretStrInput, StrInput
@@ -111,7 +111,7 @@ class AstraDBToolComponent(LCToolComponent):
         model = create_model("ToolInput", **args, __base__=BaseModel)
         return {"ToolInput": model}
 
-    def build_tool(self) -> StructuredTool:
+    def build_tool(self) -> Tool:
         """Builds an Astra DB Collection tool.
 
         Returns:
diff --git a/src/backend/base/langflow/components/tools/astradb_cql.py b/src/backend/base/langflow/components/tools/astradb_cql.py
index 04f1da824..652f4db52 100644
--- a/src/backend/base/langflow/components/tools/astradb_cql.py
+++ b/src/backend/base/langflow/components/tools/astradb_cql.py
@@ -4,7 +4,7 @@ from typing import Any
 
 import requests
 from langchain.pydantic_v1 import BaseModel, Field, create_model
-from langchain_core.tools import StructuredTool
+from langchain_core.tools import StructuredTool, Tool
 
 from langflow.base.langchain_utilities.model import LCToolComponent
 from langflow.io import DictInput, IntInput, SecretStrInput, StrInput
@@ -140,7 +140,7 @@ class AstraDBCQLToolComponent(LCToolComponent):
         model = create_model("ToolInput", **args, __base__=BaseModel)
         return {"ToolInput": model}
 
-    def build_tool(self) -> StructuredTool:
+    def build_tool(self) -> Tool:
         """Builds a Astra DB CQL Table tool.
 
         Args: