diff --git a/tests/LLM.py b/tests/LLM.py
index 578723ce..57360294 100644
--- a/tests/LLM.py
+++ b/tests/LLM.py
@@ -3,7 +3,7 @@ import os
 from unittest.mock import patch, MagicMock
 from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain
 
-from your_module import LLM  # import the module where you defined the LLM class
+from swarms.utils.llm import LLM
 
 class TestLLM(unittest.TestCase):
     @patch.object(HuggingFaceHub, '__init__', return_value=None)