EntityStateMachine now extends StateMachine<>