Change return type of method in model class