Try to fix knn test.