diff --git a/tests/test_models/test_restorers/test_basic_restorer.py b/tests/test_models/test_restorers/test_basic_restorer.py
index 78e415e021..30f4eb9d19 100644
--- a/tests/test_models/test_restorers/test_basic_restorer.py
+++ b/tests/test_models/test_restorers/test_basic_restorer.py
@@ -140,7 +140,7 @@ def test_basic_restorer():
         # evaluation with metrics must have gt images
         restorer(lq=inputs, test_mode=True)
 
-    if version.parse(torch.__version__) <= version.parse('1.5.1'):
+    if version.parse(torch.__version__) > version.parse('1.5.1'):
         with tempfile.TemporaryDirectory() as tmpdir:
             outputs = restorer(
                 **data_batch,