diff --git a/src/backend/catalog/pg_constraint.c b/src/backend/catalog/pg_constraint.c
index 2d5ac1ea813..59ae9befff4 100644
--- a/src/backend/catalog/pg_constraint.c
+++ b/src/backend/catalog/pg_constraint.c
@@ -28,6 +28,7 @@
 #include "catalog/pg_operator.h"
 #include "catalog/pg_type.h"
 #include "commands/defrem.h"
+#include "commands/extension.h"
 #include "common/int.h"
 #include "utils/array.h"
 #include "utils/builtins.h"
@@ -393,6 +394,13 @@ CreateConstraintEntry(const char *constraintName,
 	InvokeObjectPostCreateHookArg(ConstraintRelationId, conOid, 0,
 								  is_internal);
 
+	/*
+	 * If the constraint is being created as part of an extension script,
+	 * mark it as a member of the extension.
+	 */
+	if (!is_internal)
+		recordDependencyOnCurrentExtension(&conobject, false);
+
 	return conOid;
 }