fixed minor end game bug