diff --git a/YOPO/loss/guidance_loss.py b/YOPO/loss/guidance_loss.py
index 66e111b..e8f00b2 100644
--- a/YOPO/loss/guidance_loss.py
+++ b/YOPO/loss/guidance_loss.py
@@ -26,9 +26,10 @@ class GuidanceLoss(nn.Module):
         traj_dir = end_pos - cur_pos  # [B, 3]
         goal_dir = goal - cur_pos  # [B, 3]
 
-        # NOTE: distance_loss performs better in general tasks, while we choose terminal_aware_similarity_loss only for higher speed in large-scale scenario.
-        # guidance_loss = self.distance_loss(traj_dir, goal_dir)
-        guidance_loss = self.terminal_aware_similarity_loss(traj_dir, goal_dir)
+        # NOTE: trajectory with distance_loss is straighter and reaches the goal more accurately,
+        #       while our pre-trained model choose terminal_aware_similarity_loss only for higher speed in large-scale scenario.
+        guidance_loss = self.distance_loss(traj_dir, goal_dir)
+        # guidance_loss = self.terminal_aware_similarity_loss(traj_dir, goal_dir)
         return guidance_loss
 
     def distance_loss(self, traj_dir, goal_dir):