diff --git a/fastlib/u/dongryel/local_polynomial_regression/dense_lpr.h b/fastlib/u/dongryel/local_polynomial_regression/dense_lpr.h
index b805c8856b..1afbbde3d3 100644
--- a/fastlib/u/dongryel/local_polynomial_regression/dense_lpr.h
+++ b/fastlib/u/dongryel/local_polynomial_regression/dense_lpr.h
@@ -770,7 +770,7 @@ class DenseLpr {
 	all_knn.ComputeNeighbors(&resulting_neighbors, &distances);
 	
 	for(index_t i = 0; i < distances.size(); i += knns) {
-	  kernels_[i / knns].Init(sqrt(distances[i]));
+	  kernels_[i / knns].Init(sqrt(distances[i + knns - 1]));
 	}
       }
     }
diff --git a/fastlib/u/dongryel/local_polynomial_regression/naive_lpr.h b/fastlib/u/dongryel/local_polynomial_regression/naive_lpr.h
index 4c8b909547..7085131be6 100644
--- a/fastlib/u/dongryel/local_polynomial_regression/naive_lpr.h
+++ b/fastlib/u/dongryel/local_polynomial_regression/naive_lpr.h
@@ -400,7 +400,7 @@ class NaiveLpr {
       all_knn.ComputeNeighbors(&resulting_neighbors, &distances);
 
       for(index_t i = 0; i < distances.size(); i += knns) {
-	kernels_[i / knns].Init(sqrt(distances[i]));
+	kernels_[i / knns].Init(sqrt(distances[i + knns - 1]));
       }
     }
   }