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 97134831e11..59826f87112 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,7 +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",
+        api_key=SecretStr("test-key"),
         model_kwargs={"max_completion_tokens": 300},
     )
     messages = [HumanMessage("Hello")]
@@ -185,3 +185,4 @@ def test_chat_completions_api_uses_model_name() -> None:
     assert payload["model"] == "gpt-5"
     assert "messages" in payload  # Chat Completions API uses 'messages'
     assert "input" not in payload
+