Fix in KNN