Fix multiclass models.