diff --git a/PhotoMakerNode.py b/PhotoMakerNode.py
index 53d168e..e2d9664 100644
--- a/PhotoMakerNode.py
+++ b/PhotoMakerNode.py
@@ -280,7 +280,12 @@ def generate_image(self, style_name, style_strength_ratio, steps, batch_size, se
             img_tensor = img_tensor.unsqueeze(0).permute(0, 2, 3, 1)
             images_tensors.append(img_tensor)
 
-        return images_tensors
+        if len(images_tensors) > 1:
+            output_image = torch.cat(images_tensors, dim=0)
+        else:
+            output_image = images_tensors[0]
+
+        return (output_image,)
 
 
 NODE_CLASS_MAPPINGS = {