diff --git a/include/linux/mem-buf.h b/include/linux/mem-buf.h
index fbea24da2ed4..2deb80bdc9f9 100644
--- a/include/linux/mem-buf.h
+++ b/include/linux/mem-buf.h
@@ -8,6 +8,8 @@
 
 #include <linux/err.h>
 #include <linux/errno.h>
+#include <linux/haven/hh_rm_drv.h>
+#include <linux/types.h>
 #include <uapi/linux/mem-buf.h>
 
 /**