initialize Schedules if it doesn't exist

JoJoJet · Apr 12, 2023 · 0e61de6 · 0e61de6
1 parent 70333b1
commit 0e61de6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/crates/bevy_ecs/src/world/mod.rs b/crates/bevy_ecs/src/world/mod.rs
@@ -1751,7 +1751,9 @@ impl World {
         label: &dyn ScheduleLabel,
         f: impl FnOnce(&mut World, &mut Schedule) -> R,
     ) -> Result<R, ScheduleNotFoundError> {
-        let Some((extracted_label, mut schedule)) = self.resource_mut::<Schedules>().remove_entry(label) else {
+        let Some((extracted_label, mut schedule))
+            = self.get_resource_or_insert_with(Schedules::default).remove_entry(label)
+        else {
             return Err(ScheduleNotFoundError(label.dyn_clone()));
         };