Bug correction to knnImp