diff --git a/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala b/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala
index 5b283299ece55..99a64d7fb4f7c 100644
--- a/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala
+++ b/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala
@@ -196,11 +196,11 @@ class GlutenClickHouseRSSColumnarShuffleAQESuite
       val res = spark.sql("select spark_partition_id(), id from t group by id").collect()
       assert(res.length == 3)
       assert(res(0).getInt(0) == 0)
-      assert(res(0).getInt(1) == 0)
+      assert(res(0).getInt(1) == 2)
       assert(res(1).getInt(0) == 1)
       assert(res(1).getInt(1) == 1)
       assert(res(2).getInt(0) == 2)
-      assert(res(2).getInt(1) == 2)
+      assert(res(2).getInt(1) == 0)
     }
   }
 }