diff --git a/tests/langchain_lab/langgraph/test_marketing_analysis_assistant.py b/tests/langchain_lab/langgraph/test_marketing_analysis_assistant.py
index e20d72a..b5a1885 100644
--- a/tests/langchain_lab/langgraph/test_marketing_analysis_assistant.py
+++ b/tests/langchain_lab/langgraph/test_marketing_analysis_assistant.py
@@ -14,7 +14,7 @@ class TestMarketingAnalysisAssistant(TestCase):
     def test_main(self):
         assistant = MarketingAnalysisAssistant(openai_api_base=os.environ["OPENAI_API_BASE"],
                                                openai_api_key=os.environ["OPENAI_API_KEY"],
-                                               model_name="gpt-4o",
+                                               model_name=os.environ["MODEL_NAME"],
                                                recursion_limit=50)
         assistant.run()