diff --git a/lerobot/common/envs/simxarm/simxarm/__init__.py b/lerobot/common/envs/simxarm/simxarm/__init__.py
index 870eb5a9..9906ed8d 100644
--- a/lerobot/common/envs/simxarm/simxarm/__init__.py
+++ b/lerobot/common/envs/simxarm/simxarm/__init__.py
@@ -4,6 +4,7 @@ import gym
 import numpy as np
 from gym.wrappers import TimeLimit
 
+from lerobot.common.envs.simxarm.simxarm.task.base import Base as Base
 from lerobot.common.envs.simxarm.simxarm.task.lift import Lift
 from lerobot.common.envs.simxarm.simxarm.task.peg_in_box import PegInBox
 from lerobot.common.envs.simxarm.simxarm.task.push import Push
diff --git a/pyproject.toml b/pyproject.toml
index 5f6c9456..d1cdd055 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -89,7 +89,7 @@ exclude = [
 
 [tool.ruff.lint]
 select = ["E4", "E7", "E9", "F", "I", "N", "B", "C4", "SIM"]
-
+ignore-init-module-imports = true
 
 [tool.poetry-dynamic-versioning]
 enable = true