diff --git a/code/aind_auto_train/plot/curriculum.py b/code/aind_auto_train/plot/curriculum.py
index aac6432..d0619a8 100644
--- a/code/aind_auto_train/plot/curriculum.py
+++ b/code/aind_auto_train/plot/curriculum.py
@@ -43,7 +43,7 @@ def draw_diagram_rules(curriculum):
     # Add nodes (stages)
     for stage in stages:
         dot.node(name=stage.name,
-                 label=stage.name,
+                 label=f'{stage.name}\n{curriculum.parameters[stage].task}',
                  shape='ellipse',
                  style='filled',
                  fillcolor=stage_color_mapper[stage.name],