Fix for JFA-StateMachine control