diff --git a/litellm/llms/custom_httpx/azure_dall_e_2.py b/litellm/llms/custom_httpx/azure_dall_e_2.py
index f361ede5bf..a6726eb98c 100644
--- a/litellm/llms/custom_httpx/azure_dall_e_2.py
+++ b/litellm/llms/custom_httpx/azure_dall_e_2.py
@@ -1,4 +1,8 @@
-import time, json, httpx, asyncio
+import asyncio
+import json
+import time
+
+import httpx
 
 
 class AsyncCustomHTTPTransport(httpx.AsyncHTTPTransport):
@@ -7,15 +11,18 @@ class AsyncCustomHTTPTransport(httpx.AsyncHTTPTransport):
     """
 
     async def handle_async_request(self, request: httpx.Request) -> httpx.Response:
-        if "images/generations" in request.url.path and request.url.params[
-            "api-version"
-        ] in [  # dall-e-3 starts from `2023-12-01-preview` so we should be able to avoid conflict
-            "2023-06-01-preview",
-            "2023-07-01-preview",
-            "2023-08-01-preview",
-            "2023-09-01-preview",
-            "2023-10-01-preview",
-        ]:
+        _api_version = request.url.params.get("api-version", "")
+        if (
+            "images/generations" in request.url.path
+            and _api_version
+            in [  # dall-e-3 starts from `2023-12-01-preview` so we should be able to avoid conflict
+                "2023-06-01-preview",
+                "2023-07-01-preview",
+                "2023-08-01-preview",
+                "2023-09-01-preview",
+                "2023-10-01-preview",
+            ]
+        ):
             request.url = request.url.copy_with(
                 path="/openai/images/generations:submit"
             )
@@ -77,15 +84,18 @@ class CustomHTTPTransport(httpx.HTTPTransport):
         self,
         request: httpx.Request,
     ) -> httpx.Response:
-        if "images/generations" in request.url.path and request.url.params[
-            "api-version"
-        ] in [  # dall-e-3 starts from `2023-12-01-preview` so we should be able to avoid conflict
-            "2023-06-01-preview",
-            "2023-07-01-preview",
-            "2023-08-01-preview",
-            "2023-09-01-preview",
-            "2023-10-01-preview",
-        ]:
+        _api_version = request.url.params.get("api-version", "")
+        if (
+            "images/generations" in request.url.path
+            and _api_version
+            in [  # dall-e-3 starts from `2023-12-01-preview` so we should be able to avoid conflict
+                "2023-06-01-preview",
+                "2023-07-01-preview",
+                "2023-08-01-preview",
+                "2023-09-01-preview",
+                "2023-10-01-preview",
+            ]
+        ):
             request.url = request.url.copy_with(
                 path="/openai/images/generations:submit"
             )
diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py
index 7c59611d73..5f22e88a17 100644
--- a/litellm/tests/test_router.py
+++ b/litellm/tests/test_router.py
@@ -1081,7 +1081,7 @@ async def test_aimg_gen_on_router():
             {
                 "model_name": "dall-e-3",
                 "litellm_params": {
-                    "model": "dall-e-3",
+                    "model": "openai/dall-e-3",
                 },
             },
             {
@@ -1137,7 +1137,7 @@ def test_img_gen_on_router():
     try:
         model_list = [
             {
-                "model_name": "dall-e-3",
+                "model_name": "openai/dall-e-3",
                 "litellm_params": {
                     "model": "dall-e-3",
                 },