diff --git a/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py b/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py
index e1dace06386..3a951c2eb77 100644
--- a/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py
+++ b/libs/partners/openai/tests/unit_tests/chat_models/test_azure.py
@@ -90,6 +90,7 @@ def test_max_completion_tokens_in_payload() -> None:
         azure_deployment="o1-mini",
         api_version="2024-12-01-preview",
         azure_endpoint="my-base-url",
+        api_key="test-key",
         model_kwargs={"max_completion_tokens": 300},
     )
     messages = [HumanMessage("Hello")]
@@ -185,3 +186,4 @@ def test_chat_completions_api_uses_model_name() -> None:
 
 
 
+