diff --git a/promptimize/prompt_cases.py b/promptimize/prompt_cases.py
index 6f61889..4444e92 100644
--- a/promptimize/prompt_cases.py
+++ b/promptimize/prompt_cases.py
@@ -1,7 +1,7 @@
 import os
 from typing import Any, Callable, List, Optional, Union
 
-from langchain.llms import OpenAI
+from langchain_community.llms import OpenAI
 from langchain.callbacks import get_openai_callback
 
 from box import Box
diff --git a/requirements.txt b/requirements.txt
index e67d69c..bb52360 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,6 +5,7 @@ click
 pygments
 pyyaml
 langchain
+langchain-community
 tabulate
 pandas
 openai