Fixed a small problem with the classifier testing function