diff --git a/litellm/proxy/image_endpoints/endpoints.py b/litellm/proxy/image_endpoints/endpoints.py
index dcbc2222d1c..0be72dd6f1a 100644
--- a/litellm/proxy/image_endpoints/endpoints.py
+++ b/litellm/proxy/image_endpoints/endpoints.py
@@ -135,7 +135,7 @@ async def image_generation(
             data["model"] = litellm.model_alias_map[data["model"]]
 
         ### LOGGING OBJECT SETUP ### - ensures router rejected requests are dispatched to failure callbacks
-        logging_obj, data = litellm.utils.function_setup(
+        logging_obj, data = litellm.utils.function_setup(  # rebind-ok: function_setup returns updated data dict
             original_function="aimage_generation",
             rules_obj=litellm.utils.Rules(),
             start_time=start_time,
diff --git a/tests/unit/proxy/test_proxy_server.py b/tests/unit/proxy/test_proxy_server.py
index 2f2563b9ee4..c4d36b68149 100644
--- a/tests/unit/proxy/test_proxy_server.py
+++ b/tests/unit/proxy/test_proxy_server.py
@@ -1004,6 +1004,7 @@ def test_img_gen(mock_aimage_generation, client_no_auth):
             size="1024x1024",
             imageConfig={"aspectRatio": "9:16", "imageSize": "1K"},
             litellm_call_id=mock.ANY,
+            litellm_logging_obj=mock.ANY,
             metadata=mock.ANY,
             proxy_server_request=mock.ANY,
             secret_fields=mock.ANY,