diff --git a/pg_variables.c b/pg_variables.c
index 9c571d7..d87662c 100644
--- a/pg_variables.c
+++ b/pg_variables.c
@@ -2953,7 +2953,11 @@ _PG_init(void)
 							 NULL,
 							 NULL);
 
+#if defined(PGPRO_EE) && (PG_VERSION_NUM >= 150000)
+	PgproRegisterXactCallback(pgvTransCallback, NULL, XACT_EVENT_KIND_VANILLA | XACT_EVENT_KIND_ATX);
+#else
 	RegisterXactCallback(pgvTransCallback, NULL);
+#endif
 	RegisterSubXactCallback(pgvSubTransCallback, NULL);
 
 	/* Install hooks. */