diff --git a/litellm/llms/openai.py b/litellm/llms/openai.py
index 91a79fa5795..3265b230f3f 100644
--- a/litellm/llms/openai.py
+++ b/litellm/llms/openai.py
@@ -711,7 +711,7 @@ class OpenAIChatCompletion(BaseLLM):
                 original_response=response,
             )
             # return response
-            return convert_to_model_response_object(response_object=model_dump(), model_response_object=model_response, response_type="image_generation")  # type: ignore
+            return convert_to_model_response_object(response_object=response.model_dump(), model_response_object=model_response, response_type="image_generation")  # type: ignore
         except OpenAIError as e:
             exception_mapping_worked = True
             raise e