diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py
index 5e1a4e2a4..3f93b3d65 100644
--- a/tests/fixtures/dataset_factories.py
+++ b/tests/fixtures/dataset_factories.py
@@ -26,7 +26,6 @@ import pytest
 import torch
 from datasets import Dataset
 
-from lerobot.datasets.video_utils import encode_video_frames
 from lerobot.datasets.lerobot_dataset import CODEBASE_VERSION, LeRobotDataset, LeRobotDatasetMetadata
 from lerobot.datasets.utils import (
     DEFAULT_CHUNK_SIZE,
@@ -39,6 +38,7 @@ from lerobot.datasets.utils import (
     get_hf_features_from_features,
     hf_transform_to_torch,
 )
+from lerobot.datasets.video_utils import encode_video_frames
 from tests.fixtures.constants import (
     DEFAULT_FPS,
     DUMMY_CAMERA_FEATURES,
diff --git a/tests/policies/test_policies.py b/tests/policies/test_policies.py
index 57ac0edc1..3a8c6a224 100644
--- a/tests/policies/test_policies.py
+++ b/tests/policies/test_policies.py
@@ -23,6 +23,9 @@ import torch
 from safetensors.torch import load_file
 
 from lerobot import available_policies
+from lerobot.configs.default import DatasetConfig
+from lerobot.configs.train import TrainPipelineConfig
+from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature
 from lerobot.datasets.factory import make_dataset
 from lerobot.datasets.utils import cycle, dataset_to_policy_features
 from lerobot.envs.factory import make_env, make_env_config
@@ -37,9 +40,6 @@ from lerobot.policies.factory import (
 from lerobot.policies.normalize import Normalize, Unnormalize
 from lerobot.policies.pretrained import PreTrainedPolicy
 from lerobot.utils.random_utils import seeded_context
-from lerobot.configs.default import DatasetConfig
-from lerobot.configs.train import TrainPipelineConfig
-from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature
 from tests.artifacts.policies.save_policy_to_safetensors import get_policy_stats
 from tests.utils import DEVICE, require_cpu, require_env, require_x86_64_kernel