diff --git a/docs/love_loop.png b/docs/love_loop.png
index 6f016a5..6e9467b 100644
Binary files a/docs/love_loop.png and b/docs/love_loop.png differ
diff --git a/src/steer_heal/plot.py b/src/steer_heal/plot.py
index 2916110..7ba0303 100644
--- a/src/steer_heal/plot.py
+++ b/src/steer_heal/plot.py
@@ -83,7 +83,7 @@ def write_trajectory(run_dir: Path, stages: list[dict], primary_key: str = "care
 
     kind = [s["stage"] for s in stages]
     xi = list(range(len(stages)))
-    xlab = ["base" if k == "base" else f"r{s['round']}·{k[:5]}" for s, k in zip(stages, kind)]
+    xlab = ["base" if k == "base" else f"r{s['round']}·{k}" for s, k in zip(stages, kind)]
 
     fig = make_subplots(
         rows=2, cols=2, column_widths=[0.52, 0.48], row_heights=[0.5, 0.5],