Fix end game