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 59826f87112..37529d2b2b7 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
@@ -110,7 +110,7 @@ def test_max_tokens_and_max_completion_tokens_parameters() -> None:
         azure_deployment="gpt-35-turbo",
         api_version="2024-12-01-preview",
         azure_endpoint="my-base-url",
-        api_key="test-key",
+        api_key=SecretStr("test-key"),
         max_tokens=100,
     )
 
@@ -186,3 +186,4 @@ def test_chat_completions_api_uses_model_name() -> None:
     assert "messages" in payload  # Chat Completions API uses 'messages'
     assert "input" not in payload
 
+