diff --git a/src/backend/distributed/sql/citus--12.1-1--12.2-1.sql b/src/backend/distributed/sql/citus--12.1-1--12.2-1.sql
index 98bde35f9..84734eaee 100644
--- a/src/backend/distributed/sql/citus--12.1-1--12.2-1.sql
+++ b/src/backend/distributed/sql/citus--12.1-1--12.2-1.sql
@@ -32,5 +32,4 @@ DROP FUNCTION pg_catalog.citus_internal_add_colocation_metadata(int, int, int, r
 DROP FUNCTION pg_catalog.citus_internal_add_object_metadata(text, text[], text[], integer, integer, boolean);
 #include "udfs/citus_internal_add_object_metadata/12.2-1.sql"
 
-DROP FUNCTION pg_catalog.citus_internal_add_partition_metadata(regclass, "char", text, integer, "char");
 #include "udfs/citus_internal_add_partition_metadata/12.2-1.sql"
diff --git a/src/test/regress/expected/multi_extension.out b/src/test/regress/expected/multi_extension.out
index 2c3a7a51c..bab08cb00 100644
--- a/src/test/regress/expected/multi_extension.out
+++ b/src/test/regress/expected/multi_extension.out
@@ -1424,7 +1424,6 @@ SELECT * FROM multi_extension.print_extension_changes();
 ---------------------------------------------------------------------
  function citus_internal_add_colocation_metadata(integer,integer,integer,regtype,oid) void    |
  function citus_internal_add_object_metadata(text,text[],text[],integer,integer,boolean) void |
- function citus_internal_add_partition_metadata(regclass,"char",text,integer,"char") void     |
                                                                                               | function citus_internal.acquire_citus_advisory_object_class_lock(integer,cstring) void
                                                                                               | function citus_internal.add_colocation_metadata(integer,integer,integer,regtype,oid) void
                                                                                               | function citus_internal.add_object_metadata(text,text[],text[],integer,integer,boolean) void
@@ -1434,7 +1433,7 @@ SELECT * FROM multi_extension.print_extension_changes();
                                                                                               | function citus_internal.mark_object_distributed(oid,text,oid,text) void
                                                                                               | function citus_internal.start_management_transaction(xid8) void
                                                                                               | function citus_internal_database_command(text) void
-(12 rows)
+(11 rows)
 
 DROP TABLE multi_extension.prev_objects, multi_extension.extension_diff;
 -- show running version
diff --git a/src/test/regress/expected/upgrade_list_citus_objects.out b/src/test/regress/expected/upgrade_list_citus_objects.out
index 928ac85b4..8c8a08e19 100644
--- a/src/test/regress/expected/upgrade_list_citus_objects.out
+++ b/src/test/regress/expected/upgrade_list_citus_objects.out
@@ -71,6 +71,7 @@ ORDER BY 1;
  function citus_internal.replace_isolation_tester_func()
  function citus_internal.restore_isolation_tester_func()
  function citus_internal.start_management_transaction(xid8)
+ function citus_internal_add_partition_metadata(regclass,"char",text,integer,"char")
  function citus_internal_add_placement_metadata(bigint,bigint,integer,bigint)
  function citus_internal_add_placement_metadata(bigint,integer,bigint,integer,bigint)
  function citus_internal_add_shard_metadata(regclass,bigint,"char",text,text)
@@ -349,5 +350,5 @@ ORDER BY 1;
  view citus_stat_tenants_local
  view pg_dist_shard_placement
  view time_partitions
-(339 rows)
+(340 rows)