diff --git a/lib/genalloc.c b/lib/genalloc.c
index f365d71cdc774c089cfdce89b1aec1facb3a0b05..7e85d1e37a6eafd1f1507b3ec4c2a8095216e432 100644
--- a/lib/genalloc.c
+++ b/lib/genalloc.c
@@ -35,6 +35,7 @@
 #include <linux/interrupt.h>
 #include <linux/genalloc.h>
 #include <linux/of_device.h>
+#include <linux/vmalloc.h>
 
 static inline size_t chunk_size(const struct gen_pool_chunk *chunk)
 {