diff --git a/hw4/cs285/envs/obstacles/obstacles_env.py b/hw4/cs285/envs/obstacles/obstacles_env.py
index 19f7f835..bb5ef27a 100644
--- a/hw4/cs285/envs/obstacles/obstacles_env.py
+++ b/hw4/cs285/envs/obstacles/obstacles_env.py
@@ -72,6 +72,7 @@ def do_reset(self, reset_pose, reset_vel, reset_goal=None):
 
         #clear
         self.counter = 0
+        self.plt.figure(self.fig.number)
         self.plt.clf()
 
         #return