diff --git a/src/kmain.c b/src/kmain.c
index 81f8c84..42d8ea5 100644
--- a/src/kmain.c
+++ b/src/kmain.c
@@ -1,10 +1,13 @@
+#include <memory.h>
 #include <lib.h>
 #include <serial.h>
 #include <fb.h>
+#include <shim.h>
 
-void kmain(void *info) {
+void kmain(struct boot_info *info) {
 	char buf[20];
 	*(char*)0xB8000 = 'c';
+	memory_init(info->map);
 	//*(char*)(0xB8002 + 0x20'0000) = 'd';
 	itoa((long)info, buf, 16);
 	//fb_init(1024, 768);