diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameStatSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameStatSuite.scala
index 64ec1a70c47e686edd513e7649a161ace16b7472..765094da6bda75e9be37670fdccd38c143cdb910 100644
--- a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameStatSuite.scala
+++ b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameStatSuite.scala
@@ -78,7 +78,7 @@ class DataFrameStatSuite extends SparkFunSuite  {
     val rows = crosstab.collect()
     rows.foreach { row =>
       val i = row.getString(0).toInt
-      for (col <- 1 to 9) {
+      for (col <- 1 until columnNames.length) {
         val j = columnNames(col).toInt
         assert(row.getLong(col) === expected.getOrElse((i, j), 0).toLong)
       }