diff --git a/litellm/llms/gemini/image_generation/transformation.py b/litellm/llms/gemini/image_generation/transformation.py
index 14c6955772c..e57364fd288 100644
--- a/litellm/llms/gemini/image_generation/transformation.py
+++ b/litellm/llms/gemini/image_generation/transformation.py
@@ -190,10 +190,9 @@ class GoogleImageGenConfig(BaseImageGenerationConfig):
         predictions = response_data.get("predictions", [])
         for prediction in predictions:
             # Google AI returns base64 encoded images in the prediction
-            for image_data in prediction:
-                model_response.data.append(ImageObject(
-                    b64_json=image_data.get("bytesBase64Encoded", None),
-                    url=None,  # Google AI returns base64, not URLs
-                ))
+            model_response.data.append(ImageObject(
+                b64_json=prediction.get("bytesBase64Encoded", None),
+                url=None,  # Google AI returns base64, not URLs
+            ))
         
         return model_response
\ No newline at end of file