bug if initial_state is none