diff --git a/lerobot/common/datasets/aloha.py b/lerobot/common/datasets/aloha.py
index f96d32b4..dbe6cee7 100644
--- a/lerobot/common/datasets/aloha.py
+++ b/lerobot/common/datasets/aloha.py
@@ -1,3 +1,4 @@
+import os.path as osp
 from pathlib import Path
 
 import torch
diff --git a/lerobot/common/datasets/pusht.py b/lerobot/common/datasets/pusht.py
index bc978b7a..e536de47 100644
--- a/lerobot/common/datasets/pusht.py
+++ b/lerobot/common/datasets/pusht.py
@@ -1,3 +1,4 @@
+import os.path as osp
 from pathlib import Path
 
 import torch
diff --git a/lerobot/common/datasets/xarm.py b/lerobot/common/datasets/xarm.py
index 7e69e7d7..be1c596a 100644
--- a/lerobot/common/datasets/xarm.py
+++ b/lerobot/common/datasets/xarm.py
@@ -1,3 +1,4 @@
+import os.path as osp
 from pathlib import Path
 
 import torch