diff --git a/nodes.py b/nodes.py
index e824bd5..b28ccf0 100644
--- a/nodes.py
+++ b/nodes.py
@@ -703,8 +703,7 @@ class GrowMaskWithBlur:
                 # Convert back to tensor
                 out[idx] = pil2tensor(pil_image)
         
-        blurred = torch.cat(out, dim=0)
-        print(blurred.shape)
+        blurred = torch.stack(out, dim=0)
         return (blurred, 1.0 - blurred)