diff --git a/__pycache__/efficiency_nodes.cpython-312.pyc b/__pycache__/efficiency_nodes.cpython-312.pyc
new file mode 100644
index 0000000..7e00245
Binary files /dev/null and b/__pycache__/efficiency_nodes.cpython-312.pyc differ
diff --git a/__pycache__/tsc_utils.cpython-312.pyc b/__pycache__/tsc_utils.cpython-312.pyc
new file mode 100644
index 0000000..775fcc9
Binary files /dev/null and b/__pycache__/tsc_utils.cpython-312.pyc differ
diff --git a/efficiency_nodes.py b/efficiency_nodes.py
index 286d84d..9e5471e 100644
--- a/efficiency_nodes.py
+++ b/efficiency_nodes.py
@@ -3957,6 +3957,10 @@ class TSC_ImageOverlay:
             samples = overlay_image.movedim(-1, 1)
             overlay_image = comfy.utils.common_upscale(samples, overlay_image_size[0], overlay_image_size[1], resize_method, False)
             overlay_image = overlay_image.movedim(1, -1)
+        
+        # Handle batch dimension - use first image if overlay_image is a batch
+        if len(overlay_image.shape) == 4:
+            overlay_image = overlay_image[0]
             
         overlay_image = tensor2pil(overlay_image)