Fix the classifier