diff --git a/src/page_pool.rs b/src/page_pool.rs
index 5161ae7..8636af7 100644
--- a/src/page_pool.rs
+++ b/src/page_pool.rs
@@ -9,6 +9,7 @@
 use alloc::vec::Vec;
 use spin::Mutex;
 
+use super::address::PhysicalAddress;
 use super::kernel_space::KernelSpace;
 use super::region_pool::{Region, RegionPool, RegionPoolError};
 
@@ -62,8 +63,8 @@
     }
 
     /// Get physical address
-    pub fn get_pa(&self) -> usize {
-        self.pa
+    pub fn get_pa(&self) -> PhysicalAddress {
+        PhysicalAddress(self.pa)
     }
 
     /// Get as mutable slice
@@ -210,7 +211,7 @@
         assert_eq!(area.as_ptr() as usize, pages.pa);
         assert_eq!(area.len(), pages.length);
         assert!(pages.used);
-        assert_eq!(area.as_ptr() as usize, pages.get_pa());
+        assert_eq!(PhysicalAddress(area.as_ptr() as usize), pages.get_pa());
         assert_eq!(area.as_ptr() as usize, pages.base());
         assert_eq!(area.len(), pages.length());
         assert!(pages.used());