diff --git a/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.data-00000-of-00001 b/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.data-00000-of-00001
deleted file mode 100644
index b25aa2872..000000000
Binary files a/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.data-00000-of-00001 and /dev/null differ
diff --git a/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.index b/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.index
deleted file mode 100644
index 8abcb6ea5..000000000
Binary files a/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.index and /dev/null differ
diff --git a/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.meta b/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.meta
deleted file mode 100644
index e1369a3d1..000000000
Binary files a/examples/pybullet/gym/data/agent/tf_graph_data/tf_graph_data.ckpt.meta and /dev/null differ
diff --git a/examples/pybullet/gym/minitaurGymEnvTest.py b/examples/pybullet/gym/minitaurGymEnvTest.py
index 484842102..3b1ce8aeb 100644
--- a/examples/pybullet/gym/minitaurGymEnvTest.py
+++ b/examples/pybullet/gym/minitaurGymEnvTest.py
@@ -14,10 +14,9 @@ from envs.bullet.minitaurGymEnv import MinitaurGymEnv
 try:
   import sonnet
   from agents import simpleAgentWithSonnet as agent_lib
-  ckpt_path = 'data/agent/tf_graph_data/tf_graph_data_converted.ckpt-0'
 except ImportError:
   from agents import simpleAgent as agent_lib
-  ckpt_path = 'data/agent/tf_graph_data/tf_graph_data.ckpt'
+
   
 def testSinePolicy():
   """Tests sine policy
@@ -60,7 +59,7 @@ def testDDPGPolicy():
   environment = MinitaurGymEnv(render=True)
   sum_reward = 0
   steps = 1000
-
+  ckpt_path = 'data/agent/tf_graph_data/tf_graph_data_converted.ckpt-0'
   observation_shape = (31,)
   action_size = 8
   actor_layer_size = (100, 181)