diff --git a/src/backend/distributed/citus--5.1-2--5.1-3.sql b/src/backend/distributed/citus--5.1-2--5.1-3.sql
index 2a6a8bdce..5e1cdebee 100644
--- a/src/backend/distributed/citus--5.1-2--5.1-3.sql
+++ b/src/backend/distributed/citus--5.1-2--5.1-3.sql
@@ -1,6 +1,6 @@
-DROP FUNCTION master_update_shard_statistics(shard_id bigint);
+DROP FUNCTION IF EXISTS public.master_update_shard_statistics(shard_id bigint);
 
-CREATE FUNCTION pg_catalog.master_update_shard_statistics(shard_id bigint)
+CREATE OR REPLACE FUNCTION pg_catalog.master_update_shard_statistics(shard_id bigint)
     RETURNS bigint
     LANGUAGE C STRICT
     AS 'MODULE_PATHNAME', $$master_update_shard_statistics$$;