diff --git a/utils/misc.py b/utils/misc.py
index 041b726..504f895 100644
--- a/utils/misc.py
+++ b/utils/misc.py
@@ -156,7 +156,7 @@ def save_batch_imgs(name, vis_batch, save_dir):
 def save_img_dir(out, vis_batch):
     os.makedirs(out, exist_ok=True)
     for i in range(len(vis_batch)):
-        path = f"{out}/{j:05d}.png"
+        path = f"{out}/{i:05d}.png"
         imageio.imwrite(path, vis_batch[i])