diff --git a/gym_ignition/__init__.py b/gym_ignition/__init__.py
index f1d90a946..a822039a8 100644
--- a/gym_ignition/__init__.py
+++ b/gym_ignition/__init__.py
@@ -44,8 +44,8 @@
     id='CartPoleGymppy-Discrete-v0',
     entry_point='gym_ignition.base.gazebo_env:GazeboEnv',
     max_episode_steps=5000,
-    kwargs={'task': cartpole_discrete.CartPoleDiscrete,
-            'robot': sim.cartpole.CartPoleRobot,
+    kwargs={'task_cls': cartpole_discrete.CartPoleDiscrete,
+            'robot_cls': sim.cartpole.CartPoleRobot,
             'sdf': "CartPole/CartPole.sdf",
             'world': "DefaultEmptyWorld.world",
             'rtf': max_float,
@@ -57,8 +57,8 @@
     id='CartPoleGymppy-Continuous-v0',
     entry_point='gym_ignition.base.gazebo_env:GazeboEnv',
     max_episode_steps=5000,
-    kwargs={'task': cartpole_continuous.CartPoleContinuous,
-            'robot': sim.cartpole.CartPoleRobot,
+    kwargs={'task_cls': cartpole_continuous.CartPoleContinuous,
+            'robot_cls': sim.cartpole.CartPoleRobot,
             'sdf': "CartPole/CartPole.sdf",
             'world': "DefaultEmptyWorld.world",
             'rtf': max_float,