fix end of game for zombie mode