diff --git a/src/include/imd_private.h b/src/include/imd_private.h
index 43c3b7d50b..fea7315de1 100644
--- a/src/include/imd_private.h
+++ b/src/include/imd_private.h
@@ -1,6 +1,7 @@
 #ifndef _IMD_PRIVATE_H_
 #define _IMD_PRIVATE_H_
 
+#include <cbmem.h>
 #include <commonlib/bsd/helpers.h>
 
 /* In-memory data structures. */
diff --git a/src/security/vboot/vboot_loader.c b/src/security/vboot/vboot_loader.c
index 234ae8f4ad..56a0664328 100644
--- a/src/security/vboot/vboot_loader.c
+++ b/src/security/vboot/vboot_loader.c
@@ -2,6 +2,7 @@
 
 #include <boot_device.h>
 #include <cbfs.h>
+#include <cbmem.h>
 #include <commonlib/bsd/cbfs_private.h>
 #include <console/console.h>
 #include <ec/google/chromeec/ec.h>
diff --git a/src/soc/amd/common/block/pi/refcode_loader.c b/src/soc/amd/common/block/pi/refcode_loader.c
index 9fc73d4e59..274291de56 100644
--- a/src/soc/amd/common/block/pi/refcode_loader.c
+++ b/src/soc/amd/common/block/pi/refcode_loader.c
@@ -2,6 +2,7 @@
 
 #include <acpi/acpi.h>
 #include <cbfs.h>
+#include <cbmem.h>
 #include <console/console.h>
 #include <rmodule.h>
 #include <stage_cache.h>