Changed the reference from the classifier to the context for the classifier.