Classifier small fix