Change state unit test to correctly call set_state.