fixed bug in Classifier