diff --git a/aidge_core/unit_tests/test_operator_squeeze.py b/aidge_core/unit_tests/test_operator_squeeze.py
index b43605893f32f17e7b544b2fea09b16bdd982050..52120bba8b26d082486a629df1eab5ddcc2149f5 100644
--- a/aidge_core/unit_tests/test_operator_squeeze.py
+++ b/aidge_core/unit_tests/test_operator_squeeze.py
@@ -184,9 +184,7 @@ class TestSqueeze(unittest.TestCase):
             input_data.set_backend("cpu")
             test_squeeze_op.set_input(0, input_data)
 
-            with self.assertRaises((RuntimeError, AssertionError)):
-                test_squeeze_op.forward_dims()
-                test_squeeze_op.forward()
+            self.assertFalse(test_squeeze_op.forward_dims())
         return