diff --git a/data/campaigns/Secrets_of_the_Ancients/scenarios/09_Training_Session.cfg b/data/campaigns/Secrets_of_the_Ancients/scenarios/09_Training_Session.cfg
index 7622c35d2064..61d277121133 100644
--- a/data/campaigns/Secrets_of_the_Ancients/scenarios/09_Training_Session.cfg
+++ b/data/campaigns/Secrets_of_the_Ancients/scenarios/09_Training_Session.cfg
@@ -111,6 +111,7 @@
             {TRAIT_{TRAIT2}}
         [/modifications]
         facing=sw
+        role=trainee
     [/unit]
 #enddef
 
@@ -329,12 +330,7 @@
             # If there are none of the *original* DA's left (the kids don't count)...
             [not]
                 [have_unit]
-                    type=Dark Adept SotA, Dark Sorcerer SotA, Necromancer SotA
-                    [and]
-                        [filter_wml]
-                            upkeep=full
-                        [/filter_wml]
-                    [/and]
+                    role=trainee
                 [/have_unit]
             [/not]
             [then]
@@ -421,6 +417,12 @@
         [/message]
     [/event]
 
+    [event]
+        name=scenario end
+
+        {MODIFY_UNIT (role=trainee) role ""}
+    [/event]
+
     {HERO_DEATHS}
     {KIDS_ADVANCE}
 [/scenario]