diff --git a/core/src/main/scala/org/apache/spark/Partitioner.scala b/core/src/main/scala/org/apache/spark/Partitioner.scala
index d5ab7931433b1..37053bb6f37ad 100644
--- a/core/src/main/scala/org/apache/spark/Partitioner.scala
+++ b/core/src/main/scala/org/apache/spark/Partitioner.scala
@@ -136,7 +136,7 @@ class RangePartitioner[K : Ordering : ClassTag, V](
           } else {
             // The weight is 1 over the sampling probability.
             val weight = (n.toDouble / sample.size).toFloat
-            sample.foreach { key =>
+            for (key <- sample) {
               candidates += ((key, weight))
             }
           }