diff --git a/litellm/tests/test_configs/custom_callbacks.py b/litellm/tests/test_configs/custom_callbacks.py
index 27b968ac9a..b65abcfc9f 100644
--- a/litellm/tests/test_configs/custom_callbacks.py
+++ b/litellm/tests/test_configs/custom_callbacks.py
@@ -53,7 +53,7 @@ class MyCustomHandler(CustomLogger):
         self.async_success = True
         print("Value of async success: ", self.async_success)
         print("\n kwargs: ", kwargs)
-        if kwargs.get("model") == "text-embedding-ada-002":
+        if kwargs.get("model") == "azure-embedding-model":
             self.async_success_embedding = True
             self.async_embedding_kwargs = kwargs
             self.async_embedding_response = response_obj
diff --git a/litellm/tests/test_configs/test_custom_logger.yaml b/litellm/tests/test_configs/test_custom_logger.yaml
index 9673b73bae..c4f44f9ef6 100644
--- a/litellm/tests/test_configs/test_custom_logger.yaml
+++ b/litellm/tests/test_configs/test_custom_logger.yaml
@@ -1,7 +1,16 @@
 model_list: 
-  - model_name: "litellm-test-model"
-    litellm_params: 
-      model: "gpt-3.5-turbo"
+  - model_name: Azure OpenAI GPT-4 Canada
+    litellm_params:
+      model: azure/chatgpt-v-2
+      api_base: os.environ/AZURE_API_BASE
+      api_key: os.environ/AZURE_API_KEY
+      api_version: "2023-07-01-preview"
+  - model_name: azure-embedding-model
+    litellm_params:
+      model: azure/azure-embedding-model
+      api_base: os.environ/AZURE_API_BASE
+      api_key: os.environ/AZURE_API_KEY
+      api_version: "2023-07-01-preview"
 
 litellm_settings:
   drop_params: True