diff --git a/iz_helpers/sd_helpers.py b/iz_helpers/sd_helpers.py
index 3a4b543..7b0ac0f 100644
--- a/iz_helpers/sd_helpers.py
+++ b/iz_helpers/sd_helpers.py
@@ -73,7 +73,7 @@ def renderImg2Img(
     
     processed = process_images(p)
     # For those that use Image grids this will make sure that ffmpeg does not crash out
-    if (processed.images[0].size[0] != processed.images[1].size[0]):
+    if (processed.images[0].size[0] != processed.images[-1].size[0]):
         processed.images.pop(0)
         print("\nGrid image detected applying patch")