diff --git a/tests/layers/test_mask.py b/tests/layers/test_mask.py
index bd69e64a..972d3f7f 100644
--- a/tests/layers/test_mask.py
+++ b/tests/layers/test_mask.py
@@ -76,7 +76,7 @@ def test_conv_two_estimators_even(self, img_input_even: torch.Tensor):
 
     def test_conv_error(self):
         with pytest.raises(ValueError):
-            _ = MaskedLinear(8, 2, num_estimators=1, scale=None)
+            MaskedConv2d(10, 2, num_estimators=2, kernel_size=1, scale=None)
 
         with pytest.raises(ValueError):
-            _ = MaskedLinear(10, 2, num_estimators=1, scale=0)
+            MaskedConv2d(10, 2, num_estimators=2, kernel_size=1, scale=0)
diff --git a/tests/transforms/test_corruptions.py b/tests/transforms/test_corruptions.py
index cf0cb4d1..46b07ce3 100644
--- a/tests/transforms/test_corruptions.py
+++ b/tests/transforms/test_corruptions.py
@@ -76,6 +76,7 @@ def test_gaussian_blur(self):
         transform(inputs)
         transform = GaussianBlur(0)
         transform(inputs)
+        print(transform)
 
     def test_glass_blur(self):
         with pytest.raises(ValueError):