Fixes UI issue on death