diff --git a/src/backend/base/langflow/base/embeddings/aiml_embeddings.py b/src/backend/base/langflow/base/embeddings/aiml_embeddings.py
index 694e05c03..de908e756 100644
--- a/src/backend/base/langflow/base/embeddings/aiml_embeddings.py
+++ b/src/backend/base/langflow/base/embeddings/aiml_embeddings.py
@@ -2,8 +2,8 @@ import concurrent.futures
 import json
 
 import httpx
-from langchain_core.pydantic_v1 import BaseModel, SecretStr
 from loguru import logger
+from pydantic import BaseModel, SecretStr
 
 from langflow.field_typing import Embeddings