diff --git a/fancy_gym/envs/__init__.py b/fancy_gym/envs/__init__.py
index d416dbb..a40c81f 100644
--- a/fancy_gym/envs/__init__.py
+++ b/fancy_gym/envs/__init__.py
@@ -215,6 +215,7 @@ for reward_type in ["Dense", "TemporalSparse", "TemporalSpatialSparse"]:
     register(
         id='fancy/BoxPushingRandomInit{}-v0'.format(reward_type),
         entry_point='fancy_gym.envs.mujoco:BoxPushing{}'.format(reward_type),
+        mp_wrapper=mujoco.box_pushing.MPWrapper,
         max_episode_steps=MAX_EPISODE_STEPS_BOX_PUSHING,
         kwargs={"random_init": True}
     )