Fix endgame screen