Update StateMachine.test.js