Fixed bug with multiclassification