diff --git a/src/test/regress/expected/partition_wise_join.out b/src/test/regress/expected/partition_wise_join.out
index 1b00fcef3..63ae67af3 100644
--- a/src/test/regress/expected/partition_wise_join.out
+++ b/src/test/regress/expected/partition_wise_join.out
@@ -1,5 +1,6 @@
 CREATE SCHEMA partition_wise_join;
 SET search_path to partition_wise_join;
+SET citus.next_shard_id TO 360147;
 CREATE TABLE partitioning_hash_test(id int, subid int) PARTITION BY HASH(subid);
 CREATE TABLE partitioning_hash_test_0 PARTITION OF partitioning_hash_test FOR VALUES WITH (MODULUS 3, REMAINDER 0);
 CREATE TABLE partitioning_hash_test_1 PARTITION OF partitioning_hash_test FOR VALUES WITH (MODULUS 3, REMAINDER 1);
diff --git a/src/test/regress/expected/partition_wise_join_0.out b/src/test/regress/expected/partition_wise_join_0.out
index 9bda58598..559862094 100644
--- a/src/test/regress/expected/partition_wise_join_0.out
+++ b/src/test/regress/expected/partition_wise_join_0.out
@@ -1,5 +1,6 @@
 CREATE SCHEMA partition_wise_join;
 SET search_path to partition_wise_join;
+SET citus.next_shard_id TO 360147;
 CREATE TABLE partitioning_hash_test(id int, subid int) PARTITION BY HASH(subid);
 CREATE TABLE partitioning_hash_test_0 PARTITION OF partitioning_hash_test FOR VALUES WITH (MODULUS 3, REMAINDER 0);
 CREATE TABLE partitioning_hash_test_1 PARTITION OF partitioning_hash_test FOR VALUES WITH (MODULUS 3, REMAINDER 1);
diff --git a/src/test/regress/sql/partition_wise_join.sql b/src/test/regress/sql/partition_wise_join.sql
index 38e6d7a0b..0e91240e3 100644
--- a/src/test/regress/sql/partition_wise_join.sql
+++ b/src/test/regress/sql/partition_wise_join.sql
@@ -1,6 +1,8 @@
 CREATE SCHEMA partition_wise_join;
 SET search_path to partition_wise_join;
 
+SET citus.next_shard_id TO 360147;
+
 CREATE TABLE partitioning_hash_test(id int, subid int) PARTITION BY HASH(subid);
 
 CREATE TABLE partitioning_hash_test_0 PARTITION OF partitioning_hash_test FOR VALUES WITH (MODULUS 3, REMAINDER 0);