diff --git a/birefnetNode.py b/birefnetNode.py
index 8021fdf..ce2c63d 100644
--- a/birefnetNode.py
+++ b/birefnetNode.py
@@ -204,9 +204,6 @@ def rem_bg(self, model, images):
         out_images = torch.cat(_images, dim=0)
         out_masks = torch.cat(_masks, dim=0)
 
-        if out_masks.shape[0] == 1:
-            out_masks = out_masks.squeeze(0)
-
         return out_images, out_masks