Fix classifier to pass test