diff --git a/examples/pybullet/gym/pybullet_envs/prediction/__init__.py b/examples/pybullet/gym/pybullet_envs/prediction/__init__.py
index a629e0fe7..3304d604f 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_sim.py
-from . import pybullet_sim_gym_env.py
+from . import boxstack_pybullet_simy
+from . import pybullet_sim_gym_envy