diff --git a/src/backend/distributed/sql/udfs/split_shard_replication_setup/11.0-2.sql b/src/backend/distributed/sql/udfs/split_shard_replication_setup/11.0-2.sql
index 65234142e..7089466ee 100644
--- a/src/backend/distributed/sql/udfs/split_shard_replication_setup/11.0-2.sql
+++ b/src/backend/distributed/sql/udfs/split_shard_replication_setup/11.0-2.sql
@@ -1,7 +1,7 @@
 CREATE OR REPLACE FUNCTION pg_catalog.split_shard_replication_setup(
-    shardInfo integer[][])
+    shardInfo bigint[][])
 RETURNS bigint
 LANGUAGE C STRICT
 AS 'MODULE_PATHNAME', $$split_shard_replication_setup$$;
-COMMENT ON FUNCTION pg_catalog.split_shard_replication_setup(shardInfo integer[][])
+COMMENT ON FUNCTION pg_catalog.split_shard_replication_setup(shardInfo bigint[][])
     IS 'Replication setup for splitting a shard'
\ No newline at end of file
diff --git a/src/backend/distributed/sql/udfs/split_shard_replication_setup/latest.sql b/src/backend/distributed/sql/udfs/split_shard_replication_setup/latest.sql
index 65234142e..7089466ee 100644
--- a/src/backend/distributed/sql/udfs/split_shard_replication_setup/latest.sql
+++ b/src/backend/distributed/sql/udfs/split_shard_replication_setup/latest.sql
@@ -1,7 +1,7 @@
 CREATE OR REPLACE FUNCTION pg_catalog.split_shard_replication_setup(
-    shardInfo integer[][])
+    shardInfo bigint[][])
 RETURNS bigint
 LANGUAGE C STRICT
 AS 'MODULE_PATHNAME', $$split_shard_replication_setup$$;
-COMMENT ON FUNCTION pg_catalog.split_shard_replication_setup(shardInfo integer[][])
+COMMENT ON FUNCTION pg_catalog.split_shard_replication_setup(shardInfo bigint[][])
     IS 'Replication setup for splitting a shard'
\ No newline at end of file