diff --git a/src/backend/access/nbtree/nbtree.c b/src/backend/access/nbtree/nbtree.c
index 558ace1..ba01874 100644
*** a/src/backend/access/nbtree/nbtree.c
--- b/src/backend/access/nbtree/nbtree.c
***************
*** 31,36 ****
--- 31,37 ----
  #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 e02f008..fd0e86a 100644
*** a/src/backend/access/nbtree/nbtsort.c
--- b/src/backend/access/nbtree/nbtsort.c
***************
*** 70,75 ****
--- 70,76 ----
  #include "access/nbtree.h"
  #include "miscadmin.h"
  #include "storage/smgr.h"
+ #include "tcop/tcopprot.h"
  #include "utils/rel.h"
  #include "utils/tuplesort.h"