diff --git a/src/arch/x86_64/gdt.rs b/src/arch/x86_64/gdt.rs
index f5c950cd9f965a8389e4659bc59a2b15dda0e12d..c1713143ae094f93c16da977872f7f21b11ddfe8 100644
--- a/src/arch/x86_64/gdt.rs
+++ b/src/arch/x86_64/gdt.rs
@@ -99,7 +99,7 @@ pub unsafe fn set_tcb(pid: usize) {
 
 #[cfg(feature = "pti")]
 pub unsafe fn set_tss_stack(stack: usize) {
-    use arch::x86_64::pti::{PTI_CPU_STACK, PTI_CONTEXT_STACK};
+    use super::pti::{PTI_CPU_STACK, PTI_CONTEXT_STACK};
     TSS.rsp[0] = (PTI_CPU_STACK.as_ptr() as usize + PTI_CPU_STACK.len()) as u64;
     PTI_CONTEXT_STACK = stack;
 }
diff --git a/src/arch/x86_64/pti.rs b/src/arch/x86_64/pti.rs
index 336371557377f46aec3c93203abd0818817d214e..8312521b706af63afa2f3bd8f3049c5ac343e351 100644
--- a/src/arch/x86_64/pti.rs
+++ b/src/arch/x86_64/pti.rs
@@ -2,11 +2,11 @@
 use core::ptr;
 
 #[cfg(feature = "pti")]
-use memory::Frame;
+use crate::memory::Frame;
 #[cfg(feature = "pti")]
-use paging::ActivePageTable;
+use crate::paging::ActivePageTable;
 #[cfg(feature = "pti")]
-use paging::entry::EntryFlags;
+use crate::paging::entry::EntryFlags;
 
 #[cfg(feature = "pti")]
 #[thread_local]