diff --git a/src/backend/distributed/planner/intermediate_result_pruning.c b/src/backend/distributed/planner/intermediate_result_pruning.c
index 474d767d7..d4d9c0e21 100644
--- a/src/backend/distributed/planner/intermediate_result_pruning.c
+++ b/src/backend/distributed/planner/intermediate_result_pruning.c
@@ -12,6 +12,8 @@
  *-------------------------------------------------------------------------
  */
 
+#include "postgres.h"
+
 #include "common/hashfn.h"
 #include "utils/builtins.h"
 
diff --git a/src/backend/distributed/utils/replication_origin_session_utils.c b/src/backend/distributed/utils/replication_origin_session_utils.c
index 63038c0ee..eaf64d2cc 100644
--- a/src/backend/distributed/utils/replication_origin_session_utils.c
+++ b/src/backend/distributed/utils/replication_origin_session_utils.c
@@ -8,6 +8,8 @@
  *-------------------------------------------------------------------------
  */
 
+#include "postgres.h"
+
 #include "miscadmin.h"
 
 #include "utils/builtins.h"