diff --git a/src/lerobot/datasets/pyav_utils.py b/src/lerobot/datasets/pyav_utils.py
index 20ec44e2e..a4ad60e5e 100644
--- a/src/lerobot/datasets/pyav_utils.py
+++ b/src/lerobot/datasets/pyav_utils.py
@@ -28,7 +28,7 @@ from typing import TYPE_CHECKING, Any
 import av
 
 if TYPE_CHECKING:
-    from lerobot.datasets.video_utils import VideoEncoderConfig
+    from .video_utils import VideoEncoderConfig
 
 logger = logging.getLogger(__name__)
 
diff --git a/src/lerobot/datasets/video_utils.py b/src/lerobot/datasets/video_utils.py
index b8d85b6b8..5064d61f7 100644
--- a/src/lerobot/datasets/video_utils.py
+++ b/src/lerobot/datasets/video_utils.py
@@ -37,7 +37,7 @@ import torchvision
 from datasets.features.features import register_feature
 from PIL import Image
 
-from lerobot.datasets.pyav_utils import (
+from .pyav_utils import (
     check_video_encoder_config_pyav,
     detect_available_encoders_pyav,
 )