diff --git a/src/brevitas/fx/__init__.py b/src/brevitas/fx/__init__.py
index 50e66a159..325f2dea4 100644
--- a/src/brevitas/fx/__init__.py
+++ b/src/brevitas/fx/__init__.py
@@ -7,7 +7,7 @@
 from brevitas import torch_version
 
 # This needs to be bumped until https://github.com/pytorch/pytorch/pull/94461 gets merged
-if torch_version < version.parse('2.2'):
+if torch_version < version.parse('2.5'):
     import brevitas.backport.fx as fx
     from brevitas.backport.fx._compatibility import compatibility
     from brevitas.backport.fx._symbolic_trace import _assert_is_none