diff --git a/compass_image_loader.py b/compass_image_loader.py
index 164a556..d13efcb 100644
--- a/compass_image_loader.py
+++ b/compass_image_loader.py
@@ -128,8 +128,8 @@ class CompassImageLoader:
             },
         }
 
-    RETURN_TYPES = ("IMAGE", "STRING", "INT", "INT", "INT")
-    RETURN_NAMES = ("IMAGE", "path", "width", "height", "frame_count")
+    RETURN_TYPES = ("IMAGE", "STRING", "STRING", "INT", "INT", "INT")
+    RETURN_NAMES = ("IMAGE", "path", "direction", "width", "height", "frame_count")
     FUNCTION = "load_images"
 
     def load_images(
@@ -181,7 +181,7 @@ class CompassImageLoader:
             tensors[0] if len(tensors) == 1 else torch.cat(tensors, dim=0)
         )
 
-        return (image_batch, output_path, final_w, final_h, len(selected_files))
+        return (image_batch, output_path, direction, final_w, final_h, len(selected_files))
 
     @classmethod
     def IS_CHANGED(