diff --git a/aitbc/training_setup/stage_runner.py b/aitbc/training_setup/stage_runner.py
index 14522032..c50c1273 100644
--- a/aitbc/training_setup/stage_runner.py
+++ b/aitbc/training_setup/stage_runner.py
@@ -83,7 +83,7 @@ class StageRunner:
         
         return StageDefinition(
             stage=data['stage'],
-            title=data['title'],
+            title=data.get('title', f"Stage {data['stage']}"),
             commands=commands,
             expected=expected,
             prerequisites=data.get('prerequisites')