Better test for KNN