diff --git a/tests/local_testing/test_custom_callback_input.py b/tests/local_testing/test_custom_callback_input.py
index 36fa31f4ac..cd73045cb2 100644
--- a/tests/local_testing/test_custom_callback_input.py
+++ b/tests/local_testing/test_custom_callback_input.py
@@ -774,7 +774,9 @@ async def test_async_embedding_openai():
         customHandler_failure = CompletionCustomHandler()
         litellm.callbacks = [customHandler_success]
         response = await litellm.aembedding(
-            model="azure/text-embedding-ada-002",
+            model="text-embedding-ada-002",
+            input=["good morning from litellm"],
+        )
         await asyncio.sleep(1)
         print(f"customHandler_success.errors: {customHandler_success.errors}")
         print(f"customHandler_success.states: {customHandler_success.states}")