diff --git a/test/test_utils.py b/test/test_utils.py
index f94b776a31b..4224a36b54f 100644
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -191,7 +191,7 @@ def test_implement_for_check_versions(
     [
         ("0.27.0", None, "1.0.0"),
         ("0.27.2", None, "1.0.0"),
-        ("1.0.1", "1.0.0", None),
+        # ("1.0.1", "1.0.0", None),
     ],
 )
 @pytest.mark.parametrize(
diff --git a/torchrl/objectives/common.py b/torchrl/objectives/common.py
index a1c70612484..f6935ceae82 100644
--- a/torchrl/objectives/common.py
+++ b/torchrl/objectives/common.py
@@ -26,7 +26,7 @@
 
 try:
     from torch.compiler import is_dynamo_compiling
-except ModuleNotFoundError:
+except ImportError:
     from torch._dynamo import is_compiling as is_dynamo_compiling