diff --git a/examples/pybullet/gym/pybullet_envs/scene_abstract.py b/examples/pybullet/gym/pybullet_envs/scene_abstract.py
index cc74839c8..b50dd524b 100644
--- a/examples/pybullet/gym/pybullet_envs/scene_abstract.py
+++ b/examples/pybullet/gym/pybullet_envs/scene_abstract.py
@@ -39,7 +39,7 @@ class Scene:
         """
         return not self.multiplayer
 
-    def episode_restart(self):
+    def episode_restart(self, bullet_client):
         "This function gets overridden by specific scene, to reset specific objects into their start positions"
         self.cpp_world.clean_everything()
         #self.cpp_world.test_window_history_reset()