diff --git a/ir_gdb.c b/ir_gdb.c
index a231f56..7272bc8 100644
--- a/ir_gdb.c
+++ b/ir_gdb.c
@@ -7,6 +7,7 @@
  * Based on Mike Pall's implementation of GDB interface for LuaJIT.
  */
 
+#include <stddef.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <fcntl.h>
diff --git a/ir_private.h b/ir_private.h
index 802c6f0..107f395 100644
--- a/ir_private.h
+++ b/ir_private.h
@@ -8,7 +8,7 @@
 #ifndef IR_PRIVATE_H
 #define IR_PRIVATE_H
 #include <string.h>
-#include <malloc.h>
+#include <stdlib.h>
 
 #ifdef IR_DEBUG
 # include <assert.h>