Fix crash at startup

benfrankel · Dec 30, 2023 · 9bb44d5 · 9bb44d5
1 parent 3e9ba5e
commit 9bb44d5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/state/editor_screen/upgrade_panel.rs b/src/state/editor_screen/upgrade_panel.rs
@@ -38,7 +38,12 @@ impl Plugin for UpgradePanelPlugin {
                         .and_then(state_changed::<AppState>().or_else(on_event::<UpgradeEvent>())),
                 ),
             )
-            .add_systems(Update, update_upgrade_button_disabled.in_set(AppSet::End));
+            .add_systems(
+                Update,
+                update_upgrade_button_disabled
+                    .in_set(AppSet::End)
+                    .run_if(in_state(AppState::EditorScreen)),
+            );
     }
 }
 

diff --git a/src/upgrade.rs b/src/upgrade.rs
@@ -51,7 +51,7 @@ impl Plugin for UpgradePlugin {
             .add_systems(
                 Update,
                 (
-                    process_new_installed_upgrades,
+                    process_new_installed_upgrades.run_if(on_event::<UpgradeEvent>()),
                     install_upgrades.run_if(on_event::<UpgradeEvent>()),
                     run_installed_upgrades,
                     apply_deferred,