diff --git a/tests/models/test_LLM.py b/tests/models/test_LLM.py
index 04d6a5f2..44bc29c9 100644
--- a/tests/models/test_LLM.py
+++ b/tests/models/test_LLM.py
@@ -1,7 +1,8 @@
 import unittest
 import os
 from unittest.mock import patch
-from langchain import HuggingFaceHub, ChatOpenAI
+from langchain import HuggingFaceHub
+from langchain.chat_models import ChatOpenAI
 
 from swarms.models.llm import LLM