Fix specs to reflect introducing state_machine