diff --git a/.flake8 b/.flake8
index 0d4b9398ff..27c8693a5c 100644
--- a/.flake8
+++ b/.flake8
@@ -8,6 +8,6 @@ select =
     # Missing argument descriptions in the docstring
     D417,
     # TorchFix
-    TOR
+    TOR0,TOR1,TOR2
 max-line-length = 120
 exclude = docs/source,third_party
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index fb071843f4..ef0fcd12b3 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -34,6 +34,6 @@ repos:
       - id: flake8
         additional_dependencies:
           - flake8-docstrings == 1.6.0
-          - torchfix == 0.0.1
+          - torchfix == 0.0.2
         args:
           - --config=.flake8