diff --git a/src/backend/langflow/components/llms/ChatOllamaEndpoint.py b/src/backend/langflow/components/llms/ChatOllamaEndpoint.py
index 3201ff432..34e9e0bd7 100644
--- a/src/backend/langflow/components/llms/ChatOllamaEndpoint.py
+++ b/src/backend/langflow/components/llms/ChatOllamaEndpoint.py
@@ -1,8 +1,8 @@
 from typing import Any, Dict, List, Optional
 
 # from langchain_community.chat_models import ChatOllama
-from langchain.chat_models import ChatOllama
-from langchain.chat_models.base import BaseChatModel
+from langchain_community.chat_models import ChatOllama
+from langchain_core.language_models.chat_models import BaseChatModel
 
 # from langchain.chat_models import ChatOllama
 from langflow import CustomComponent