fix initial_state setter