Move state-related methods to stateful mixin