diff --git a/hw5/meta/train_policy.py b/hw5/meta/train_policy.py
index b16a155f6..81ed30cc0 100644
--- a/hw5/meta/train_policy.py
+++ b/hw5/meta/train_policy.py
@@ -358,7 +358,7 @@ def sample_trajectory(self, env, animate_this_episode, is_evaluation):
         env.reset_task(is_evaluation=is_evaluation)
         stats = []
         #====================================================================================#
-        #                           ----------PROBLEM 2----------
+        #                           ----------PROBLEM 1----------
         #====================================================================================#
         ep_steps = 0
         steps = 0