diff --git a/comfy/cmd/server.py b/comfy/cmd/server.py
index 5fe7fc26e..3e13c2348 100644
--- a/comfy/cmd/server.py
+++ b/comfy/cmd/server.py
@@ -662,7 +662,7 @@ class PromptServer(ExecutorToClientProgress):
                     output_images.append(image_tuple)
 
             if len(output_images) > 0:
-                main_image = output_images[-1]
+                main_image = output_images[0]
                 filename = main_image["filename"]
                 digest_headers_ = {
                     "Digest": f"SHA-256={content_digest}",