diff --git a/src/lerobot/policies/wall_x/__init__.py b/src/lerobot/policies/wall_x/__init__.py
index d80c27bda89..16fd2c8ab89 100644
--- a/src/lerobot/policies/wall_x/__init__.py
+++ b/src/lerobot/policies/wall_x/__init__.py
@@ -15,5 +15,7 @@
 # limitations under the License.
 
 from .configuration_wall_x import WallXConfig
+from .modeling_wall_x import WallXPolicy
+from .processor_wall_x import make_wall_x_pre_post_processors
 
 __all__ = ["WallXConfig", "WallXPolicy", "make_wall_x_pre_post_processors"]