load_state fix