One method call not changed after Model refactoring