Use state rather than initial_state