diff --git a/litellm/llms/openai.py b/litellm/llms/openai.py
index efb7ee8f9..0299c502c 100644
--- a/litellm/llms/openai.py
+++ b/litellm/llms/openai.py
@@ -642,7 +642,7 @@ class OpenAIChatCompletion(BaseLLM):
                 additional_args={"complete_input_dict": data},
                 original_response=stringified_response,
             )
-            return convert_to_model_response_object(response_object=json.loads(stringified_response), model_response_object=model_response, response_type="embedding")  # type: ignore
+            return convert_to_model_response_object(response_object=json.loads(stringified_response), model_response_object=model_response, response_type="image_generation")  # type: ignore
         except Exception as e:
             ## LOGGING
             logging_obj.post_call(