fix __set_state__