diff --git a/scripts/img2img_k.py b/scripts/img2img_k.py
index d35aef7b8..d26fce595 100644
--- a/scripts/img2img_k.py
+++ b/scripts/img2img_k.py
@@ -279,7 +279,7 @@ def main():
                         x_samples_ddim = accelerator.gather(x_samples_ddim)
 
                         if accelerator.is_main_process and not opt.skip_save:
-                            for x_sample in x_samples:
+                            for x_sample in x_samples_ddim:
                                 x_sample = 255. * rearrange(x_sample.cpu().numpy(), 'c h w -> h w c')
                                 Image.fromarray(x_sample.astype(np.uint8)).save(
                                     os.path.join(sample_path, f"{base_count:05}.png"))