diff --git a/src/ui/battle_ui.rs b/src/ui/battle_ui.rs index 34cd379..3013b58 100644 --- a/src/ui/battle_ui.rs +++ b/src/ui/battle_ui.rs @@ -49,3 +49,9 @@ pub fn spawn_battle_ui(mut commands: Commands, request: Res, fight Ok(()) } + +pub fn cleanup_battle_ui(mut commands: Commands, battle_ui: Query>) { + for entity in battle_ui { + commands.entity(entity).despawn(); + } +} \ No newline at end of file diff --git a/src/ui/mod.rs b/src/ui/mod.rs index 75b319f..f393c32 100644 --- a/src/ui/mod.rs +++ b/src/ui/mod.rs @@ -12,6 +12,7 @@ pub struct InGameUIPlugin; impl Plugin for InGameUIPlugin { fn build(&self, app: &mut App) { app.add_systems(OnEnter(AppState::InMatch), battle_ui::spawn_battle_ui) + .add_systems(OnExit(AppState::InMatch), battle_ui::cleanup_battle_ui) .add_systems(PostUpdate, fighter_info::update_fighter_info_damage); } }