Fixed a bug in  Classifier