1 files changed, 3 insertions, 6 deletions
diff --git a/kernel/memory.c b/kernel/memory.c
index 0c73970..67ad69c 100644
--- a/kernel/memory.c
+++ b/kernel/memory.c
@@ -1,3 +1,4 @@
+#include <asm/system.h>
 #include <kernel/memory.h>
 #include <stdint.h>
 #include <sys/types.h>
@@ -82,12 +83,8 @@ void *map_page(void *page) {
   page = (void*) ((uint32_t) page & ~0xFFF);
 
   ktab[(p >> 12) & 0x3FF] = (uint32_t) page | 0x003;
-  /*
-   * invalidate any cached entries for
-   * the mapped page window.
-   * TODO: skip this on i386
-   */
-  __asm__ ("invlpg %0" :: "m" (mapped_page));
+
+  invlpg(mapped_page);
 
   return mapped_page;
 }