diff --git a/src/test/regress/expected/multi_colocation_utils.out b/src/test/regress/expected/multi_colocation_utils.out
index 3a1e8ed34..a23b450fd 100644
--- a/src/test/regress/expected/multi_colocation_utils.out
+++ b/src/test/regress/expected/multi_colocation_utils.out
@@ -286,22 +286,23 @@ SELECT UNNEST(get_colocated_table_array('table6_append'))::regclass ORDER BY 1;
 (1 row)
 
 -- check co-located shard list
-SELECT get_colocated_shard_array(1300000) ORDER BY 1;
- get_colocated_shard_array 
----------------------------
- {1300000,1300004}
+SELECT UNNEST(get_colocated_shard_array(1300000))::regclass ORDER BY 1;
+ unnest  
+---------
+ 1300000
+ 1300004
+(2 rows)
+
+SELECT UNNEST(get_colocated_shard_array(1300016))::regclass ORDER BY 1;
+ unnest  
+---------
+ 1300016
 (1 row)
 
-SELECT get_colocated_shard_array(1300016) ORDER BY 1;
- get_colocated_shard_array 
----------------------------
- {1300016}
-(1 row)
-
-SELECT get_colocated_shard_array(1300020) ORDER BY 1;
- get_colocated_shard_array 
----------------------------
- {1300020}
+SELECT UNNEST(get_colocated_shard_array(1300020))::regclass ORDER BY 1;
+ unnest  
+---------
+ 1300020
 (1 row)
 
 -- check FindShardIntervalIndex function
diff --git a/src/test/regress/sql/multi_colocation_utils.sql b/src/test/regress/sql/multi_colocation_utils.sql
index 19bc09876..cf9ad6884 100644
--- a/src/test/regress/sql/multi_colocation_utils.sql
+++ b/src/test/regress/sql/multi_colocation_utils.sql
@@ -139,9 +139,9 @@ SELECT UNNEST(get_colocated_table_array('table5_groupX'))::regclass ORDER BY 1;
 SELECT UNNEST(get_colocated_table_array('table6_append'))::regclass ORDER BY 1;
 
 -- check co-located shard list
-SELECT get_colocated_shard_array(1300000) ORDER BY 1;
-SELECT get_colocated_shard_array(1300016) ORDER BY 1;
-SELECT get_colocated_shard_array(1300020) ORDER BY 1;
+SELECT UNNEST(get_colocated_shard_array(1300000))::regclass ORDER BY 1;
+SELECT UNNEST(get_colocated_shard_array(1300016))::regclass ORDER BY 1;
+SELECT UNNEST(get_colocated_shard_array(1300020))::regclass ORDER BY 1;
 
 -- check FindShardIntervalIndex function
 SELECT find_shard_interval_index(1300000);