More informative error message on missing stage (#1048)

More informative error message on missing state
bevyengine · Dec 13, 2020 · e511cdb · e511cdb
1 parent 002e22f
commit e511cdb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/bevy_ecs/src/schedule/mod.rs b/crates/bevy_ecs/src/schedule/mod.rs
@@ -158,7 +158,7 @@ impl Schedule {
     ) -> &mut Self {
         let stage = self
             .get_stage_mut::<T>(name)
-            .expect("stage does not exist or is the wrong type");
+            .unwrap_or_else(|| panic!("stage '{}' does not exist or is the wrong type", name));
         func(stage);
         self
     }