diff --git a/key_blob_utils/ocb.c b/key_blob_utils/ocb.c
index 9a68182..70172c8 100644
--- a/key_blob_utils/ocb.c
+++ b/key_blob_utils/ocb.c
@@ -73,6 +73,7 @@
 /* ----------------------------------------------------------------------- */
 
 #include <keymaster/key_blob_utils/ae.h>
+#include <malloc.h>
 #include <stdlib.h>
 #include <string.h>