correcting the classifier