diff --git a/tests/pass_through_tests/test_openai_assistants_passthrough.py b/tests/pass_through_tests/test_openai_assistants_passthrough.py
index 4736426aa43..d416b79bbf6 100644
--- a/tests/pass_through_tests/test_openai_assistants_passthrough.py
+++ b/tests/pass_through_tests/test_openai_assistants_passthrough.py
@@ -100,7 +100,8 @@ def test_openai_assistants_e2e_operations_stream():
 
 
 def test_azure_openai_assistants_e2e_operations_stream():
-    client = openai.OpenAI(
+    from openai import AzureOpenAI
+    client = AzureOpenAI(
         base_url="http://0.0.0.0:4000/azure-config-passthrough", 
         api_key="sk-1234",
         api_version="2025-01-01-preview"