refactor archHEADmain

1 files changed, 4 insertions, 2 deletions

diff --git a/src/arch/amd64/shim.c b/src/arch/amd64/shim.c
index c1c3368..05d995c 100644
--- a/src/arch/amd64/shim.c
+++ b/src/arch/amd64/shim.c
@@ -3,11 +3,12 @@
 #include <lib.h>
 #include <shim.h>
 #include <memory.h>
+#include <mboot.h>
+#include <pic.h>
 
 #include "paging.h"
-#include "mboot.h"
+#include "fpu.h"
 #include "cpu/idt.h"
-#include "drivers/pic.h"
 
 static struct boot_info boot_info;
 
@@ -16,6 +17,7 @@ void *amd64_shim(long mboot_magic, volatile void *mboot_data_ptr) {
 	paging_init();
 	pic_remap();
 	idt_init();
+	fpu_enable();
 
 	mboot_load_info(mboot_magic, mboot_data_ptr, &boot_info);