diff --git a/src/backend/distributed/sql/udfs/citus_auto_shard_split_start/12.0-1.sql b/src/backend/distributed/sql/udfs/citus_auto_shard_split_start/12.0-1.sql
index 98dc6e01a..876aaa7bc 100644
--- a/src/backend/distributed/sql/udfs/citus_auto_shard_split_start/12.0-1.sql
+++ b/src/backend/distributed/sql/udfs/citus_auto_shard_split_start/12.0-1.sql
@@ -1,4 +1,6 @@
 CREATE OR REPLACE FUNCTION pg_catalog.citus_auto_shard_split_start(
+    shard_transfer_mode citus.shard_transfer_mode default 'auto'
+
     )
     RETURNS VOID
 
@@ -6,8 +8,8 @@ CREATE OR REPLACE FUNCTION pg_catalog.citus_auto_shard_split_start(
 
     LANGUAGE C VOLATILE;
 
-COMMENT ON FUNCTION pg_catalog.citus_auto_shard_split_start()
+COMMENT ON FUNCTION pg_catalog.citus_auto_shard_split_start(citus.shard_transfer_mode)
 
     IS 'automatically split the necessary shards in the cluster in the background';
 
-GRANT EXECUTE ON FUNCTION pg_catalog.citus_auto_shard_split_start() TO PUBLIC;
+GRANT EXECUTE ON FUNCTION pg_catalog.citus_auto_shard_split_start(citus.shard_transfer_mode) TO PUBLIC;
\ No newline at end of file
diff --git a/src/backend/distributed/sql/udfs/citus_auto_shard_split_start/latest.sql b/src/backend/distributed/sql/udfs/citus_auto_shard_split_start/latest.sql
index 0750c05ea..2a1aea49f 100644
--- a/src/backend/distributed/sql/udfs/citus_auto_shard_split_start/latest.sql
+++ b/src/backend/distributed/sql/udfs/citus_auto_shard_split_start/latest.sql
@@ -1,4 +1,5 @@
 CREATE OR REPLACE FUNCTION pg_catalog.citus_auto_shard_split_start(
+    shard_transfer_mode citus.shard_transfer_mode default 'auto'
 
     )
     RETURNS VOID
@@ -7,8 +8,8 @@ CREATE OR REPLACE FUNCTION pg_catalog.citus_auto_shard_split_start(
 
     LANGUAGE C VOLATILE;
 
-COMMENT ON FUNCTION pg_catalog.citus_auto_shard_split_start()
+COMMENT ON FUNCTION pg_catalog.citus_auto_shard_split_start(citus.shard_transfer_mode)
 
     IS 'automatically split the necessary shards in the cluster in the background';
 
-GRANT EXECUTE ON FUNCTION pg_catalog.citus_auto_shard_split_start() TO PUBLIC;
+GRANT EXECUTE ON FUNCTION pg_catalog.citus_auto_shard_split_start(citus.shard_transfer_mode) TO PUBLIC;