diff --git a/urdfenvs/urdf_common/urdf_env.py b/urdfenvs/urdf_common/urdf_env.py
index f94d3b0..1393cc2 100644
--- a/urdfenvs/urdf_common/urdf_env.py
+++ b/urdfenvs/urdf_common/urdf_env.py
@@ -217,7 +217,10 @@ def step(self, action: np.ndarray):
 
         pybullet.stepSimulation(self._cid)
         for robot_id, robot in enumerate(self._robots):
-            contacts = pybullet.getContactPoints(robot._robot)
+            contacts = pybullet.getContactPoints(
+                bodyA=robot._robot,
+                physicsClientId=self._cid
+            ) or []
             for contact_info in contacts:
                 body_b = contact_info[2]
                 if body_b in self._obsts: