Fix end game screen