diff --git a/lerobot/common/datasets/abstract.py b/lerobot/common/datasets/abstract.py
index 0c0746e1..d6a51246 100644
--- a/lerobot/common/datasets/abstract.py
+++ b/lerobot/common/datasets/abstract.py
@@ -13,8 +13,7 @@ from torchrl.data.replay_buffers.replay_buffers import TensorDictReplayBuffer
 from torchrl.data.replay_buffers.samplers import SliceSampler
 from torchrl.data.replay_buffers.storages import TensorStorage, _collate_id
 from torchrl.data.replay_buffers.writers import ImmutableDatasetWriter, Writer
-
-from rl.torchrl.envs.transforms.transforms import Compose
+from torchrl.envs.transforms.transforms import Compose
 
 
 class AbstractExperienceReplay(TensorDictReplayBuffer):
diff --git a/lerobot/common/envs/factory.py b/lerobot/common/envs/factory.py
index 184646cf..c20ef441 100644
--- a/lerobot/common/envs/factory.py
+++ b/lerobot/common/envs/factory.py
@@ -1,6 +1,4 @@
-from torchrl.envs.transforms import StepCounter, TransformedEnv
-
-from rl.torchrl.envs.transforms.transforms import Compose, Transform
+from torchrl.envs.transforms import Compose, StepCounter, Transform, TransformedEnv
 
 
 def make_env(cfg, transform=None):