diff --git a/litellm/tests/test_health_check.py b/litellm/tests/test_health_check.py
index 3996c8d0986..05166b8888b 100644
--- a/litellm/tests/test_health_check.py
+++ b/litellm/tests/test_health_check.py
@@ -21,7 +21,7 @@ async def test_azure_health_check():
             "api_base": os.getenv("AZURE_API_BASE"),
             "api_version": os.getenv("AZURE_API_VERSION"),
         },
-        mode="completion",
+        mode="chat",
     )
     print(f"response: {response}")