diff --git a/src/backend/distributed/sql/udfs/citus_finish_pg_upgrade/11.1-1.sql b/src/backend/distributed/sql/udfs/citus_finish_pg_upgrade/11.1-1.sql
index 3d7e38011..caa80d51e 100644
--- a/src/backend/distributed/sql/udfs/citus_finish_pg_upgrade/11.1-1.sql
+++ b/src/backend/distributed/sql/udfs/citus_finish_pg_upgrade/11.1-1.sql
@@ -127,8 +127,8 @@ BEGIN
         'n' as deptype
     FROM pg_catalog.pg_dist_partition p;
 
-    -- disable set dependencies for columnar table access method
-    -- PERFORM citus_internal.columnar_ensure_am_depends_catalog();
+    -- set dependencies for columnar table access method
+    PERFORM columnar_internal.columnar_ensure_am_depends_catalog();
 
     -- restore pg_dist_object from the stable identifiers
     TRUNCATE pg_catalog.pg_dist_object;
diff --git a/src/backend/distributed/sql/udfs/citus_finish_pg_upgrade/latest.sql b/src/backend/distributed/sql/udfs/citus_finish_pg_upgrade/latest.sql
index 3d7e38011..caa80d51e 100644
--- a/src/backend/distributed/sql/udfs/citus_finish_pg_upgrade/latest.sql
+++ b/src/backend/distributed/sql/udfs/citus_finish_pg_upgrade/latest.sql
@@ -127,8 +127,8 @@ BEGIN
         'n' as deptype
     FROM pg_catalog.pg_dist_partition p;
 
-    -- disable set dependencies for columnar table access method
-    -- PERFORM citus_internal.columnar_ensure_am_depends_catalog();
+    -- set dependencies for columnar table access method
+    PERFORM columnar_internal.columnar_ensure_am_depends_catalog();
 
     -- restore pg_dist_object from the stable identifiers
     TRUNCATE pg_catalog.pg_dist_object;