Pass StateMachine to guard

Guards are awesome, but pretty useless without the data :)