diff --git a/Kernel/Arch/aarch64/PageDirectory.h b/Kernel/Arch/aarch64/PageDirectory.h
index 326c017f701..c05d406316f 100644
--- a/Kernel/Arch/aarch64/PageDirectory.h
+++ b/Kernel/Arch/aarch64/PageDirectory.h
@@ -38,7 +38,7 @@ constexpr u32 DEVICE_MEMORY = (1 << 2);
 class PageDirectoryEntry {
 public:
     PhysicalPtr page_table_base() const { return PhysicalAddress::physical_page_base(m_raw); }
-    void set_page_table_base(u32 value)
+    void set_page_table_base(PhysicalPtr value)
     {
         m_raw &= 0xffff000000000fffULL;
         m_raw |= PhysicalAddress::physical_page_base(value);
diff --git a/Kernel/Arch/x86/PageDirectory.h b/Kernel/Arch/x86/PageDirectory.h
index 05fa7bade04..86bc97bf8c2 100644
--- a/Kernel/Arch/x86/PageDirectory.h
+++ b/Kernel/Arch/x86/PageDirectory.h
@@ -16,7 +16,7 @@ namespace Kernel {
 class PageDirectoryEntry {
 public:
     PhysicalPtr page_table_base() const { return PhysicalAddress::physical_page_base(m_raw); }
-    void set_page_table_base(u32 value)
+    void set_page_table_base(PhysicalPtr value)
     {
         m_raw &= 0x8000000000000fffULL;
         m_raw |= PhysicalAddress::physical_page_base(value);