diff --git a/kernel/kmalloc.h b/kernel/kmalloc.h
index d88d961..d307cc0 100644
--- a/kernel/kmalloc.h
+++ b/kernel/kmalloc.h
@@ -7,6 +7,13 @@ See the file LICENSE for details.
 #ifndef KMALLOC_H
 #define KMALLOC_H
 
+struct kmalloc_chunk {
+	int state;
+	int length;
+	struct kmalloc_chunk *next;
+	struct kmalloc_chunk *prev;
+};
+
 void *kmalloc(int length);
 void kfree(void *ptr);
 
@@ -14,4 +21,8 @@ void kmalloc_init(char *start, int length);
 void kmalloc_debug();
 int kmalloc_test();
 
+#ifdef KMALLOC_EXT
+  #include "kmalloc_ext.h"
+#endif
+
 #endif