Fixing the issue of the classifier