diff --git a/src/backend/access/nbtree/nbtree.c b/src/backend/access/nbtree/nbtree.c
index 558ace1562bece64b507444b9c17aa676a8827cf..ba01874469bf7af9b18bd288e9edea5aa0be3115 100644
--- a/src/backend/access/nbtree/nbtree.c
+++ b/src/backend/access/nbtree/nbtree.c
@@ -31,6 +31,7 @@
 #include "storage/lmgr.h"
 #include "storage/predicate.h"
 #include "storage/smgr.h"
+#include "tcop/tcopprot.h"
 #include "utils/memutils.h"
 
 
diff --git a/src/backend/access/nbtree/nbtsort.c b/src/backend/access/nbtree/nbtsort.c
index e02f0081fd02b4f5286a69531c873badaa49d547..fd0e86a6aa3a19df28d523c3cacf8640763fb3bd 100644
--- a/src/backend/access/nbtree/nbtsort.c
+++ b/src/backend/access/nbtree/nbtsort.c
@@ -70,6 +70,7 @@
 #include "access/nbtree.h"
 #include "miscadmin.h"
 #include "storage/smgr.h"
+#include "tcop/tcopprot.h"
 #include "utils/rel.h"
 #include "utils/tuplesort.h"