diff --git a/src/backend/distributed/master/master_metadata_utility.c b/src/backend/distributed/master/master_metadata_utility.c
index 821574fcb..97d689a51 100644
--- a/src/backend/distributed/master/master_metadata_utility.c
+++ b/src/backend/distributed/master/master_metadata_utility.c
@@ -953,8 +953,13 @@ RecordDistributedRelationDependencies(Oid distributedRelationId, Node *distribut
 	recordDependencyOn(&relationAddr, &citusExtensionAddr, DEPENDENCY_NORMAL);
 
 	/* make sure the distribution key column/expression does not just go away */
+#if (PG_VERSION_NUM >= 100000)
+	recordDependencyOnSingleRelExpr(&relationAddr, distributionKey, distributedRelationId,
+									DEPENDENCY_NORMAL, DEPENDENCY_NORMAL, false);
+#else
 	recordDependencyOnSingleRelExpr(&relationAddr, distributionKey, distributedRelationId,
 									DEPENDENCY_NORMAL, DEPENDENCY_NORMAL);
+#endif
 }