diff --git a/modules/ImageTransform.py b/modules/ImageTransform.py
index d7d34bb..7376303 100644
--- a/modules/ImageTransform.py
+++ b/modules/ImageTransform.py
@@ -231,7 +231,7 @@ class ImageTransformCropCorners:
 
         def crop_tensor(tensor):
             return torch.stack([
-                tensor[:, :, i] - mask for i in range(tensor.shape[2])
+                (tensor[:, :, i] - mask).clamp(0, 1) for i in range(tensor.shape[2])
             ], dim=2)
 
         return (torch.stack([