diff --git a/examples/pybullet/gym/pybullet_envs/prediction/__init__.py b/examples/pybullet/gym/pybullet_envs/prediction/__init__.py
index 3304d604f..26f76e102 100644
--- a/examples/pybullet/gym/pybullet_envs/prediction/__init__.py
+++ b/examples/pybullet/gym/pybullet_envs/prediction/__init__.py
@@ -1,3 +1,3 @@
-from . import boxstack_pybullet_simy
-from . import pybullet_sim_gym_envy
+from . import boxstack_pybullet_sim
+from . import pybullet_sim_gym_env
 
diff --git a/examples/pybullet/gym/pybullet_envs/prediction/pybullet_sim_gym_env.py b/examples/pybullet/gym/pybullet_envs/prediction/pybullet_sim_gym_env.py
index 5c1b52757..eabd7b749 100644
--- a/examples/pybullet/gym/pybullet_envs/prediction/pybullet_sim_gym_env.py
+++ b/examples/pybullet/gym/pybullet_envs/prediction/pybullet_sim_gym_env.py
@@ -190,7 +190,7 @@ class PyBulletSimGymEnv(gym.Env):
     proj_matrix=[1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, -1.0000200271606445, -1.0, 0.0, 0.0, -0.02000020071864128, 0.0]
     (_, _, px, _, _) = self._pybullet_client.getCameraImage(
         width=self._render_width, height=self._render_height, viewMatrix=view_matrix,
-        projectionMatrix=proj_matrix, renderer=pybullet.ER_TINY_RENDERER)
+        projectionMatrix=proj_matrix, renderer=pybullet.ER_BULLET_HARDWARE_OPENGL)#ER_TINY_RENDERER)
     rgb_array = np.array(px, dtype=np.uint8)
     rgb_array = np.reshape(rgb_array, (self._render_height, self._render_width, 4))
     rgb_array = rgb_array[:, :, :3]