diff --git a/tests/local_testing/test_embedding.py b/tests/local_testing/test_embedding.py
index d0383f7002..9a63c6c83c 100644
--- a/tests/local_testing/test_embedding.py
+++ b/tests/local_testing/test_embedding.py
@@ -198,8 +198,8 @@ def _azure_ai_image_mock_response(*args, **kwargs):
     [
         (
             "azure_ai/Cohere-embed-v3-multilingual-2",
-            os.getenv("AZURE_AI_COHERE_API_BASE"),
-            os.getenv("AZURE_AI_COHERE_API_KEY"),
+            os.getenv("AZURE_AI_API_BASE"),
+            os.getenv("AZURE_AI_API_KEY"),
         )
     ],
 )
diff --git a/tests/vector_store_tests/test_azure_ai_vector_store.py b/tests/vector_store_tests/test_azure_ai_vector_store.py
index 52eb6635a9..58e45f259a 100644
--- a/tests/vector_store_tests/test_azure_ai_vector_store.py
+++ b/tests/vector_store_tests/test_azure_ai_vector_store.py
@@ -17,10 +17,10 @@ async def test_basic_search_vector_store(sync_mode):
         "vector_store_id": "my-vector-index",
         "custom_llm_provider": "azure_ai",
         "azure_search_service_name": "azure-kb-search",
-        "litellm_embedding_model": "azure/text-embedding-3-large",
+        "litellm_embedding_model": "azure_ai/text-embedding-3-large",
         "litellm_embedding_config": {
-            "api_base": os.getenv("AZURE_AI_SEARCH_EMBEDDING_API_BASE"),
-            "api_key": os.getenv("AZURE_AI_SEARCH_EMBEDDING_API_KEY"),
+            "api_base": os.getenv("AZURE_AI_API_BASE"),
+            "api_key": os.getenv("AZURE_AI_API_KEY"),
         },
         "api_key": os.getenv("AZURE_SEARCH_API_KEY"),
     }