diff --git a/src/ivfbuild.c b/src/ivfbuild.c
index cd37aad..d8d00c4 100644
--- a/src/ivfbuild.c
+++ b/src/ivfbuild.c
@@ -5,6 +5,7 @@
 #include "catalog/index.h"
 #include "ivfflat.h"
 #include "miscadmin.h"
+#include "port.h"
 #include "storage/bufmgr.h"
 
 #if PG_VERSION_NUM >= 140000
diff --git a/src/ivfkmeans.c b/src/ivfkmeans.c
index fc12ff5..ebb45b1 100644
--- a/src/ivfkmeans.c
+++ b/src/ivfkmeans.c
@@ -4,6 +4,7 @@
 
 #include "ivfflat.h"
 #include "miscadmin.h"
+#include "port.h"
 
 #if PG_VERSION_NUM >= 150000
 #include "common/pg_prng.h"