diff --git a/libdmabufheap/include/dmabuf_alloc.h b/libdmabufheap/include/dmabuf_alloc.h
index 07ec54b86091ef90aa141b6462ae1b4036bef6e8..e8233198269f8a818663f971f00135d625e90ead 100644
--- a/libdmabufheap/include/dmabuf_alloc.h
+++ b/libdmabufheap/include/dmabuf_alloc.h
@@ -18,6 +18,7 @@
 
 #include <stdlib.h>
 #include <linux/dma-buf.h>
+#include <linux/dma-heap.h>
 
 #undef LOG_TAG
 #define LOG_TAG "dmabufheap"
diff --git a/libdmabufheap/src/dmabuf_alloc.c b/libdmabufheap/src/dmabuf_alloc.c
index e20a781751ba8551ce883e7eb92df75e40e0cfcb..b283b78b6f7a47b80f4cb110a89a012d7e1a1c49 100644
--- a/libdmabufheap/src/dmabuf_alloc.c
+++ b/libdmabufheap/src/dmabuf_alloc.c
@@ -20,8 +20,6 @@
 #include <unistd.h>
 #include <fcntl.h>
 #include <sys/ioctl.h>
-#include <linux/dma-buf.h>
-#include <linux/dma-heap.h>
 #include "securec.h"
 #include "hilog/log.h"
 #include "dmabuf_alloc.h"