diff --git a/mace/core/preallocated_pooled_allocator.h b/mace/core/preallocated_pooled_allocator.h
index ad0c975a16ec9056b4b48512803e8e5f2a45eedd..0299d2f4cf83d89f4871806e6b504985a345e374 100644
--- a/mace/core/preallocated_pooled_allocator.h
+++ b/mace/core/preallocated_pooled_allocator.h
@@ -7,6 +7,7 @@
 
 #include <unordered_map>
 #include "mace/core/allocator.h"
+#include "mace/core/buffer.h"
 
 namespace mace {