diff --git a/Storydiffusion_node.py b/Storydiffusion_node.py
index 1058c7d..e023da2 100644
--- a/Storydiffusion_node.py
+++ b/Storydiffusion_node.py
@@ -1222,7 +1222,7 @@ def msdiffusion_main(pipe, image_1, image_2, prompts_dual, width, height, steps,
             global write, attn_count
             write = True
             attn_count = 0
-            in_img = image
+            in_img = torch.cat((in_img, image), dim=0)
             torch.cuda.empty_cache()
             if single_files:
                 try: