diff --git a/src/test/regress/expected/generated_identity.out b/src/test/regress/expected/generated_identity.out
index a857957cd..bf5e5248d 100644
--- a/src/test/regress/expected/generated_identity.out
+++ b/src/test/regress/expected/generated_identity.out
@@ -313,9 +313,10 @@ INSERT INTO color(color_name) VALUES ('Blue');
 \c - postgres - :master_port
 SET search_path TO generated_identities;
 SET client_min_messages to ERROR;
+SET citus.next_shard_id TO 1270000;
 DROP TABLE Color;
 CREATE TABLE color (
-    color_id BIGINT GENERATED ALWAYS AS IDENTITY UNIQUE,
+    color_id BIGINT GENERATED ALWAYS AS IDENTITY,
     color_name VARCHAR NOT NULL
 ) USING columnar;
 SELECT create_distributed_table('color', 'color_id');
diff --git a/src/test/regress/expected/generated_identity_0.out b/src/test/regress/expected/generated_identity_0.out
index fff175e42..7aae21f20 100644
--- a/src/test/regress/expected/generated_identity_0.out
+++ b/src/test/regress/expected/generated_identity_0.out
@@ -313,6 +313,7 @@ INSERT INTO color(color_name) VALUES ('Blue');
 \c - postgres - :master_port
 SET search_path TO generated_identities;
 SET client_min_messages to ERROR;
+SET citus.next_shard_id TO 12400000;
 DROP TABLE Color;
 CREATE TABLE color (
     color_id BIGINT GENERATED ALWAYS AS IDENTITY UNIQUE,
@@ -359,7 +360,7 @@ ERROR:  cannot insert a non-DEFAULT value into column "color_id"
 DETAIL:  Column "color_id" is an identity column defined as GENERATED ALWAYS.
 HINT:  Use OVERRIDING SYSTEM VALUE to override.
 INSERT INTO color(color_id, color_name) OVERRIDING SYSTEM VALUE VALUES (1, 'Red');
-ERROR:  duplicate key value violates unique constraint "color_color_id_key_xxxxxx"
+ERROR:  duplicate key value violates unique constraint "color_color_id_key_12400000"
 DETAIL:  Key (color_id)=(1) already exists.
 CONTEXT:  while executing command on localhost:xxxxx
 -- update null or custom value
@@ -388,7 +389,7 @@ ERROR:  cannot insert a non-DEFAULT value into column "color_id"
 DETAIL:  Column "color_id" is an identity column defined as GENERATED ALWAYS.
 HINT:  Use OVERRIDING SYSTEM VALUE to override.
 INSERT INTO color(color_id, color_name) OVERRIDING SYSTEM VALUE VALUES (1, 'Red');
-ERROR:  duplicate key value violates unique constraint "color_color_id_key_xxxxxx"
+ERROR:  duplicate key value violates unique constraint "color_color_id_key_12400000"
 DETAIL:  Key (color_id)=(1) already exists.
 CONTEXT:  while executing command on localhost:xxxxx
 -- update null or custom value
@@ -399,4 +400,4 @@ UPDATE color SET color_id = 1;
 ERROR:  column "color_id" can only be updated to DEFAULT
 DETAIL:  Column "color_id" is an identity column defined as GENERATED ALWAYS.
 DROP SCHEMA generated_identities CASCADE;
-DROP USER identity_test_user;
\ No newline at end of file
+DROP USER identity_test_user;
diff --git a/src/test/regress/sql/generated_identity.sql b/src/test/regress/sql/generated_identity.sql
index ad43ee695..6bef6a147 100644
--- a/src/test/regress/sql/generated_identity.sql
+++ b/src/test/regress/sql/generated_identity.sql
@@ -179,6 +179,7 @@ INSERT INTO color(color_name) VALUES ('Blue');
 \c - postgres - :master_port
 SET search_path TO generated_identities;
 SET client_min_messages to ERROR;
+SET citus.next_shard_id TO 12400000;
 
 DROP TABLE Color;
 CREATE TABLE color (