diff --git a/tests/policies/wall_x/test_wallx.py b/tests/policies/wall_x/test_wallx.py
index f78e6f42a..2951ca8b8 100644
--- a/tests/policies/wall_x/test_wallx.py
+++ b/tests/policies/wall_x/test_wallx.py
@@ -22,6 +22,7 @@ import pytest
 
 try:
     import peft  # noqa: F401
+
     WALLX_AVAILABLE = True
 except ImportError:
     WALLX_AVAILABLE = False
@@ -34,6 +35,7 @@ pytestmark = pytest.mark.skipif(
 
 if WALLX_AVAILABLE:
     import torch
+
     from lerobot.policies.factory import make_policy_config
     from lerobot.policies.wall_x import (
         WallXConfig,