1 files changed, 5 insertions, 7 deletions
diff --git a/arch/x86_64/src/memory/mmu.cpp b/arch/x86_64/src/memory/mmu.cpp
index e15d94e..ea23278 100644
--- a/arch/x86_64/src/memory/mmu.cpp
+++ b/arch/x86_64/src/memory/mmu.cpp
@@ -1,14 +1,12 @@
-#include "x86_64/memory/mmu.hpp"
+#include "arch/memory/mmu.hpp"
 
 #include "kapi/memory.hpp"
 
-#include "x86_64/cpu/registers.hpp"
+#include "arch/cpu/registers.hpp"
 
-namespace teachos::memory::x86_64
+namespace arch::memory
 {
-  namespace cpu = cpu::x86_64;
-
-  auto tlb_flush(linear_address address) -> void
+  auto tlb_flush(kapi::memory::linear_address address) -> void
   {
     asm volatile("invlpg (%[input])" : /* no output from call */ : [input] "r"(address) : "memory");
   }
@@ -18,4 +16,4 @@ namespace teachos::memory::x86_64
     auto paging_root = cpu::cr3::read();
     cpu::cr3::write(paging_root);
   }
-}  // namespace teachos::memory::x86_64
+}  // namespace arch::memory