diff --git a/src/backend/base/langflow/components/datastax/astradb_cql.py b/src/backend/base/langflow/components/datastax/astradb_cql.py
index 8f00db720..ea86f6554 100644
--- a/src/backend/base/langflow/components/datastax/astradb_cql.py
+++ b/src/backend/base/langflow/components/datastax/astradb_cql.py
@@ -5,8 +5,8 @@ from http import HTTPStatus
 from typing import Any
 
 import requests
-from langchain.pydantic_v1 import BaseModel, Field, create_model
 from langchain_core.tools import StructuredTool, Tool
+from pydantic import BaseModel, Field, create_model
 
 from langflow.base.langchain_utilities.model import LCToolComponent
 from langflow.io import DictInput, IntInput, SecretStrInput, StrInput, TableInput