diff --git a/tests/datasets/test_aggregate.py b/tests/datasets/test_aggregate.py
index f1f441e61..4f316f80e 100644
--- a/tests/datasets/test_aggregate.py
+++ b/tests/datasets/test_aggregate.py
@@ -14,9 +14,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import torch
 from unittest.mock import patch
 
+import torch
+
 from lerobot.datasets.aggregate import aggregate_datasets
 from lerobot.datasets.lerobot_dataset import LeRobotDataset
 from tests.fixtures.constants import DUMMY_REPO_ID
diff --git a/tests/test_control_robot.py b/tests/test_control_robot.py
index 8a4663a49..374f98129 100644
--- a/tests/test_control_robot.py
+++ b/tests/test_control_robot.py
@@ -112,7 +112,7 @@ def test_record_and_replay(tmp_path):
     )
 
     record(record_cfg)
-    
+
     # Mock the revision to prevent Hub calls during replay
     with (
         patch("lerobot.datasets.lerobot_dataset.get_safe_version") as mock_get_safe_version,