Add the transition to state in there

Otherwise it's a pain to figure out