diff --git a/examples/pybullet/gym/pybullet_envs/deep_mimic/env/pybullet_deep_mimic_env.py b/examples/pybullet/gym/pybullet_envs/deep_mimic/env/pybullet_deep_mimic_env.py
index bd032ed29..dc444f17f 100644
--- a/examples/pybullet/gym/pybullet_envs/deep_mimic/env/pybullet_deep_mimic_env.py
+++ b/examples/pybullet/gym/pybullet_envs/deep_mimic/env/pybullet_deep_mimic_env.py
@@ -409,7 +409,7 @@ class PyBulletDeepMimicEnv(Env):
     if self.goal.is_hit:
       return 1
     else:
-      goalPos = self.goal.goal_data
+      goalPos = self.goal.world_pos
       distanceSquared = sum([(x - y)**2 for (x, y) in zip(goalPos, linkPos)])
       return math.exp(-4*distanceSquared)