Fixed missing method in model