diff --git a/discord_tron_client/classes/image_manipulation/pipeline.py b/discord_tron_client/classes/image_manipulation/pipeline.py
index 042bf149..e65e6758 100644
--- a/discord_tron_client/classes/image_manipulation/pipeline.py
+++ b/discord_tron_client/classes/image_manipulation/pipeline.py
@@ -281,10 +281,11 @@ def _run_pipeline(
             elif upscaler:
                 new_image = pipe(
                     prompt_embeds=prompt_embed,
+                    num_images_per_prompt=4,
                     negative_prompt_embeds=negative_embed,
                     image=image,
                     num_inference_steps=int(float(steps)),
-                ).images[0]
+                ).images
             else:
                 raise Exception(
                     "Invalid combination of parameters for image generation"