diff --git a/__init__.py b/__init__.py
index b892011..dfc9a41 100644
--- a/__init__.py
+++ b/__init__.py
@@ -95,7 +95,7 @@ class ImageSelector:
 
         if selected_index:
             print(f"ImageSelector: selected: {len(selected_index)} images")
-            return (images[selected_index, :, :, :], )
+            return (images[selected_index], )
 
         print(f"ImageSelector: selected no images, passthrough")
         return (images, )