diff --git a/executor/partition_table_test.go b/executor/partition_table_test.go
index 73520eea1792a..dd52a733de9c6 100644
--- a/executor/partition_table_test.go
+++ b/executor/partition_table_test.go
@@ -1560,7 +1560,7 @@ func (s *partitionTableSuite) TestDirectReadingWithAgg(c *C) {
 	tk.MustExec("insert into tregular2 values " + strings.Join(vals, ","))
 
 	// test range partition
-	for i := 0; i < 2000; i++ {
+	for i := 0; i < 200; i++ {
 		// select /*+ stream_agg() */ a from t where a > ? group by a;
 		// select /*+ hash_agg() */ a from t where a > ? group by a;
 		// select /*+ stream_agg() */ a from t where a in(?, ?, ?) group by a;
@@ -1592,7 +1592,7 @@ func (s *partitionTableSuite) TestDirectReadingWithAgg(c *C) {
 	}
 
 	// test hash partition
-	for i := 0; i < 2000; i++ {
+	for i := 0; i < 200; i++ {
 		// select /*+ stream_agg() */ a from t where a > ? group by a;
 		// select /*+ hash_agg() */ a from t where a > ? group by a;
 		// select /*+ stream_agg() */ a from t where a in(?, ?, ?) group by a;
@@ -1624,7 +1624,7 @@ func (s *partitionTableSuite) TestDirectReadingWithAgg(c *C) {
 	}
 
 	// test list partition
-	for i := 0; i < 2000; i++ {
+	for i := 0; i < 200; i++ {
 		// select /*+ stream_agg() */ a from t where a > ? group by a;
 		// select /*+ hash_agg() */ a from t where a > ? group by a;
 		// select /*+ stream_agg() */ a from t where a in(?, ?, ?) group by a;