diff --git a/tests/e2e/multigpu/patched/test_sp.py b/tests/e2e/multigpu/patched/test_sp.py
index ba0636879..a005e6742 100644
--- a/tests/e2e/multigpu/patched/test_sp.py
+++ b/tests/e2e/multigpu/patched/test_sp.py
@@ -94,7 +94,7 @@ class TestSequenceParallelism:
 
         check_tensorboard(
             temp_dir + "/runs",
-            "train/loss",
+            "train/train_loss",
             threshold,
             "Train Loss (%s) is too high",
         )