diff --git a/tests/local_testing/test_custom_callback_input.py b/tests/local_testing/test_custom_callback_input.py
index e711344f654..7f4da72e04f 100644
--- a/tests/local_testing/test_custom_callback_input.py
+++ b/tests/local_testing/test_custom_callback_input.py
@@ -930,10 +930,8 @@ def test_image_generation_openai():
 
         response = litellm.image_generation(
             prompt="A cute baby sea otter",
-            model="azure/dall-e-3-test",
-            api_version="2023-12-01-preview",
-            api_base=os.getenv("AZURE_SWEDEN_API_BASE"),
-            api_key=os.getenv("AZURE_SWEDEN_API_KEY"),
+            model="openai/dall-e-3",
+            api_key=os.getenv("OPENAI_API_KEY"),
         )
 
         print(f"response: {response}")