Fix load state problem