Ok ignore my comment on the test, no need for a state machine here a simple boolean flag will do