diff --git a/platform/mem.h b/platform/mem.h
index eaac886..edaf164 100644
--- a/platform/mem.h
+++ b/platform/mem.h
@@ -1,2 +1,6 @@
 #include <string.h>
-#include <malloc.h>
+#if defined(__linux__)
+  #include <malloc.h>
+#elif defined(__APPLE__)
+  #include <stdlib.h>
+#endif