diff --git a/libgralloc/alloc_controller.cpp b/libgralloc/alloc_controller.cpp
index 1a5fe25..05ea620 100644
--- a/libgralloc/alloc_controller.cpp
+++ b/libgralloc/alloc_controller.cpp
@@ -33,7 +33,9 @@
 #include "gralloc_priv.h"
 #include "alloc_controller.h"
 #include "memalloc.h"
+#ifdef USE_ION
 #include "ionalloc.h"
+#endif
 #include "pmemalloc.h"
 #include "ashmemalloc.h"
 #include "gr.h"
diff --git a/libgralloc/pmemalloc.h b/libgralloc/pmemalloc.h
index 4aed0b1..1847971 100644
--- a/libgralloc/pmemalloc.h
+++ b/libgralloc/pmemalloc.h
@@ -30,7 +30,9 @@
 #ifndef GRALLOC_PMEMALLOC_H
 #define GRALLOC_PMEMALLOC_H
 
+#ifdef USE_ION
 #include <linux/ion.h>
+#endif
 #include <utils/RefBase.h>
 #include "memalloc.h"