fix KNN