diff --git a/__pycache__/__init__.cpython-311.pyc b/__pycache__/__init__.cpython-311.pyc
new file mode 100644
index 0000000..13bc79a
Binary files /dev/null and b/__pycache__/__init__.cpython-311.pyc differ
diff --git a/__pycache__/compass_image_loader.cpython-311.pyc b/__pycache__/compass_image_loader.cpython-311.pyc
new file mode 100644
index 0000000..28860e7
Binary files /dev/null and b/__pycache__/compass_image_loader.cpython-311.pyc differ
diff --git a/compass_image_loader.py b/compass_image_loader.py
index f142cd0..880c490 100644
--- a/compass_image_loader.py
+++ b/compass_image_loader.py
@@ -105,7 +105,7 @@ class CompassImageLoader:
     RETURN_TYPES = ("IMAGE", "STRING", "STRING", "INT", "INT", "INT")
     RETURN_NAMES = ("IMAGE", "path", "direction", "width", "height", "frame_count")
     FUNCTION = "load_images"
-    OUTPUT_NODE = False
+    OUTPUT_NODE = True
 
     def load_images(
         self, directory, direction, modality, frame=None, width=0, height=0,
@@ -164,7 +164,7 @@ class CompassImageLoader:
             tensors[0] if len(tensors) == 1 else torch.cat(tensors, dim=0)
         )
 
-        return (image_batch, output_path, resolved_direction, final_w, final_h, len(selected_files))
+        return {"ui": {"images": [{"filename": os.path.basename(output_path), "subfolder": "", "type": "input"}]}, "result": (image_batch, output_path, resolved_direction, final_w, final_h, len(selected_files))}
 
     @classmethod
     def IS_CHANGED(