diff --git a/concordia/language_model/langchain_ollama_model.py b/concordia/language_model/langchain_ollama_model.py
index e1b6059..e08aa83 100644
--- a/concordia/language_model/langchain_ollama_model.py
+++ b/concordia/language_model/langchain_ollama_model.py
@@ -19,7 +19,7 @@
 from concordia.language_model import language_model
 from concordia.utils import measurements as measurements_lib
 from concordia.utils import sampling
-from langchain.llms import ollama
+from langchain_community.llms import ollama
 
 from typing_extensions import override
 
diff --git a/setup.py b/setup.py
index 1c287af..0c08874 100644
--- a/setup.py
+++ b/setup.py
@@ -72,7 +72,7 @@ def _remove_excluded(description: str) -> str:
         'google-cloud-aiplatform',
         'google-generativeai',
         'ipython',
-        'langchain',
+        'langchain-community',
         'matplotlib',
         'mistralai',
         'numpy',