diff --git a/src/scripts/app.ts b/src/scripts/app.ts
index 9da1166122..0f82b85081 100644
--- a/src/scripts/app.ts
+++ b/src/scripts/app.ts
@@ -379,8 +379,8 @@ export class ComfyApp {
       selectedIndex = node.imageIndex
     }
 
-    const paintedIndex = selectedIndex + 1
-    const combinedIndex = selectedIndex + 2
+    const paintedIndex = imgs ? imgs.length + 1 : 1
+    const combinedIndex = imgs ? imgs.length + 2 : 2
 
     // for vueNodes mode
     const images =