diff --git a/src/nodes/helpers.py b/src/nodes/helpers.py
index 84570da..a7a6d90 100644
--- a/src/nodes/helpers.py
+++ b/src/nodes/helpers.py
@@ -313,7 +313,8 @@ def load_images_wrapper(file_names, embed_transparency=False, disp_names=None, s
                 all_preview_imgs.append(get_image_preview_info(file_name))
 
     logger.debug(f"Loaded {len(masks)} batches:")
-    for n, i in enumerate(masks):
+    items = masks if is_mask else imgs
+    for n, i in enumerate(items):
         logger.debug(f"{n}) {i.shape}")
 
     result = (imgs, masks, used_file_names) if channel is None else (masks, used_file_names)