summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/game/unit_soldier.cpp1
-rw-r--r--src/game/units.cpp2
2 files changed, 2 insertions, 1 deletions
diff --git a/src/game/unit_soldier.cpp b/src/game/unit_soldier.cpp
index 7985df6..0ffe19f 100644
--- a/src/game/unit_soldier.cpp
+++ b/src/game/unit_soldier.cpp
@@ -228,7 +228,6 @@ void unit_soldier_t::on_death(void)
cmodel.cflags = CF_BACKGROUND;
place(world, x);
controllable = false;
- game->selected_units.erase(this);
move_marker.reset();
aim_marker.reset();
}
diff --git a/src/game/units.cpp b/src/game/units.cpp
index 735ed89..f1ee45e 100644
--- a/src/game/units.cpp
+++ b/src/game/units.cpp
@@ -237,6 +237,8 @@ void unit_t::die(unit_t *killer)
sleep();
ignore_waking = true;
+ game->selected_units.erase(this);
+
on_death();
}