Fix the KNN Bug