fix multiclass generation