diff --git a/src/hnswbuild.c b/src/hnswbuild.c
index c677a47..12d789e 100644
--- a/src/hnswbuild.c
+++ b/src/hnswbuild.c
@@ -53,6 +53,7 @@
 #include "storage/bufmgr.h"
 #include "storage/bufpage.h"
 #include "storage/condition_variable.h"
+#include "storage/dsm.h"
 #include "storage/lockdefs.h"
 #include "storage/lwlock.h"
 #include "storage/shm_toc.h"
diff --git a/src/hnswinsert.c b/src/hnswinsert.c
index 945bd84..63636f8 100644
--- a/src/hnswinsert.c
+++ b/src/hnswinsert.c
@@ -6,8 +6,10 @@
 #include "storage/bufmgr.h"
 #include "storage/bufpage.h"
 #include "storage/lmgr.h"
+#include "storage/lockdefs.h"
 #include "utils/datum.h"
 #include "utils/memutils.h"
+#include "utils/rel.h"
 
 #if PG_VERSION_NUM >= 160000
 #include "varatt.h"
diff --git a/src/hnswscan.c b/src/hnswscan.c
index 45f7021..4c2c884 100644
--- a/src/hnswscan.c
+++ b/src/hnswscan.c
@@ -6,6 +6,7 @@
 #include "access/skey.h"
 #include "hnsw.h"
 #include "lib/pairingheap.h"
+#include "miscadmin.h"
 #include "nodes/pg_list.h"
 #include "pgstat.h"
 #include "storage/lmgr.h"
diff --git a/src/hnswutils.c b/src/hnswutils.c
index 91b773d..f127b4e 100644
--- a/src/hnswutils.c
+++ b/src/hnswutils.c
@@ -13,6 +13,7 @@
 #include "sparsevec.h"
 #include "storage/bufmgr.h"
 #include "storage/bufpage.h"
+#include "storage/lwlock.h"
 #include "utils/datum.h"
 #include "utils/memdebug.h"
 #include "utils/rel.h"
diff --git a/src/hnswvacuum.c b/src/hnswvacuum.c
index dbbe0a9..e53252b 100644
--- a/src/hnswvacuum.c
+++ b/src/hnswvacuum.c
@@ -14,6 +14,7 @@
 #include "storage/lockdefs.h"
 #include "storage/off.h"
 #include "utils/memutils.h"
+#include "utils/rel.h"
 
 #if PG_VERSION_NUM >= 160000
 #include "varatt.h"
diff --git a/src/ivfbuild.c b/src/ivfbuild.c
index 48e4eb5..1d18fb3 100644
--- a/src/ivfbuild.c
+++ b/src/ivfbuild.c
@@ -22,6 +22,7 @@
 #include "miscadmin.h"
 #include "optimizer/optimizer.h"
 #include "storage/bufmgr.h"
+#include "storage/bufpage.h"
 #include "storage/condition_variable.h"
 #include "storage/lockdefs.h"
 #include "storage/shm_toc.h"