Fixed compilation error in the model class