Resolve assert on TestKnn.