diff --git a/tests/test_torchfix.py b/tests/test_torchfix.py
index 03c0bd0..9a4896f 100644
--- a/tests/test_torchfix.py
+++ b/tests/test_torchfix.py
@@ -36,7 +36,7 @@ def pytest_generate_tests(metafunc):
             ("TOR102", {"TOR102"}),
             ("TOR102,TOR101", {"TOR102", "TOR101"}),
             ("TOR1,TOR102", {"TOR102", "TOR101", "TOR103", "TOR104", "TOR105"}),
-            (None, set(GET_ALL_ERROR_CODES() - exclude_set)),
+            (None, set(GET_ALL_ERROR_CODES()) - exclude_set),
         ]
         metafunc.parametrize("case,expected", cases, ids=[case for case, _ in cases])