diff --git a/tensordict/_td.py b/tensordict/_td.py
index 7c2ff332f..fd95707fa 100644
--- a/tensordict/_td.py
+++ b/tensordict/_td.py
@@ -121,7 +121,7 @@
             is_batchedtensor,
         )
     except ImportError:
-        from functorch._C import is_batchedtensor  # @manual=fbcode//functorch:_C
+        from functorch._C import is_batchedtensor  # @manual=fbcode//caffe2/functorch:_C
 
     _has_functorch = True
 except ImportError:
diff --git a/tensordict/utils.py b/tensordict/utils.py
index 65f0ef4c4..81f0ecb00 100644
--- a/tensordict/utils.py
+++ b/tensordict/utils.py
@@ -82,7 +82,7 @@
             is_batchedtensor,
         )
     except ImportError:
-        from functorch._C import (  # @manual=fbcode//functorch:_C  # noqa
+        from functorch._C import (  # @manual=fbcode//caffe2/functorch:_C  # noqa
             get_unwrapped,
             is_batchedtensor,
         )