diff --git a/classy_vision/tasks/classification_task.py b/classy_vision/tasks/classification_task.py
index d95cb7826..eaf7200ba 100644
--- a/classy_vision/tasks/classification_task.py
+++ b/classy_vision/tasks/classification_task.py
@@ -1337,7 +1337,7 @@ def log_phase_end(self, tag):
         self.perf_log.append(
             {
                 "tag": tag,
-                "phase_idx": self.train_phase_idx,
+                "phase_idx": self.train_phase_idx if self.train else self.eval_phase_idx,
                 "im_per_sec": im_per_sec,
             }
         )