Fixed bug in classifier