diff --git a/tests/test_litellm/llms/azure/test_azure_common_utils.py b/tests/test_litellm/llms/azure/test_azure_common_utils.py
index 1c81f12e0fe..0e5ddb391ee 100644
--- a/tests/test_litellm/llms/azure/test_azure_common_utils.py
+++ b/tests/test_litellm/llms/azure/test_azure_common_utils.py
@@ -438,6 +438,8 @@ def test_select_azure_base_url_called(setup_mocks):
             "allm_passthrough_route",
             "llm_passthrough_route",
             "asearch",
+            "avector_store_create",
+            "avector_store_search",
         ]
     ],
 )