Fix set_state preinit stuff