fix minor possible restore state issue