Override helper methods for State Machine to work