diff --git a/torchrl/envs/batched_envs.py b/torchrl/envs/batched_envs.py
index 068d37f8135..02c7f5893dc 100644
--- a/torchrl/envs/batched_envs.py
+++ b/torchrl/envs/batched_envs.py
@@ -22,13 +22,13 @@
 import torch
 
 from tensordict import (
-    _zip_strict,
     is_tensor_collection,
     LazyStackedTensorDict,
     TensorDict,
     TensorDictBase,
     unravel_key,
 )
+from tensordict.utils import _zip_strict
 from torch import multiprocessing as mp
 from torchrl._utils import (
     _check_for_faulty_process,