diff --git a/nodes.py b/nodes.py
index e711502..e672493 100644
--- a/nodes.py
+++ b/nodes.py
@@ -41,7 +41,7 @@ class PadAndResize:
     
     def resize(self, image, latent, mode):
         b, c, sh, sw = image.shape
-        _, _, th, tw = latent['samples'].shape
+        *_, th, tw = latent['samples'].shape
         th, tw = th * 8, tw * 8
         if (th, tw) == (sh, sw):
             return image