diff --git a/tests/test_multi_thresholding.py b/tests/test_multi_thresholding.py
index 58ac0332a6f7d4fe143d4cffdf6f4da874aad344..0d3c8292129b9793201c4f083f15f715238b7262 100644
--- a/tests/test_multi_thresholding.py
+++ b/tests/test_multi_thresholding.py
@@ -220,7 +220,4 @@ def test_execute_multi_thresholding():
         ),
     )
 
-    if outputs == results:
-        return True
-    else:
-        return False
+    assert (outputs == results).all()