diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
index 683a48c5ff..c1948d89f1 100755
--- a/ci/gpu/build.sh
+++ b/ci/gpu/build.sh
@@ -200,7 +200,7 @@ else
 
     # FIXME: Project FLASH only builds for python version 3.8 which is the one used in
     # the CUDA 11.0 job, need to change all versions to project flash
-    if [ "$PYTHON" == "3.8" ];then
+    if [ "$CUDA_REL" == "11.0" ];then
         gpuci_logger "Using Project FLASH to install cuml python"
         CONDA_FILE=`find ${CONDA_ARTIFACT_PATH} -name "cuml*.tar.bz2"`
         CONDA_FILE=`basename "$CONDA_FILE" .tar.bz2` #get filename without extension
diff --git a/python/cuml/test/test_nearest_neighbors.py b/python/cuml/test/test_nearest_neighbors.py
index dfc8711711..acbb7dfb1d 100644
--- a/python/cuml/test/test_nearest_neighbors.py
+++ b/python/cuml/test/test_nearest_neighbors.py
@@ -547,14 +547,8 @@ def test_nearest_neighbors_rbc(distance, n_neighbors, nrows):
         brute_d, brute_i = knn_cu_brute.kneighbors(
             X[:query_rows, :], n_neighbors=n_neighbors)
 
-    rbc_i = cp.sort(rbc_i, axis=1)
-    brute_i = cp.sort(brute_i, axis=1)
-
-    # TODO: These are failing with 1 or 2 mismatched elements
-    # for very small values of k:
-    # https://github.com/rapidsai/cuml/issues/4262
-    assert len(brute_d[brute_d != rbc_d]) <= 3
-    assert len(brute_i[brute_i != rbc_i]) <= 3
+    assert len(brute_d[brute_d != rbc_d]) == 0
+    assert len(brute_i[brute_i != rbc_i]) == 0
 
 
 @pytest.mark.parametrize("metric", valid_metrics_sparse())