diff --git a/litellm/tests/test_exceptions.py b/litellm/tests/test_exceptions.py
index fa62b1970ce..16f7ae65d48 100644
--- a/litellm/tests/test_exceptions.py
+++ b/litellm/tests/test_exceptions.py
@@ -37,7 +37,7 @@ def test_context_window(model):
     try:
         azure = model == "chatgpt-test"
         print(f"model: {model}")
-        response = completion(model=model, messages=messages, custom_llm_provider=custom_llm_provider, logger_fn=logging_fn)
+        response = completion(model=model, messages=messages, custom_llm_provider="azure", logger_fn=logging_fn)
         print(f"response: {response}")
     except InvalidRequestError:
         print("InvalidRequestError")