Updated unit selection on death