diff --git a/example.py b/example.py
index cef271c6..a6e78dcc 100644
--- a/example.py
+++ b/example.py
@@ -1,7 +1,7 @@
-from langchain.models import ChatOpenAI
+from langchain.models import OpenAIChat
 from swarms import Worker
 
-llm = ChatOpenAI(
+llm = OpenAIChat(
     model_name='gpt-4', 
     openai_api_key="api-key", 
     temperature=0.5