diff --git a/src/efi/mod.rs b/src/efi/mod.rs
index 1046112d..afbb3d8c 100644
--- a/src/efi/mod.rs
+++ b/src/efi/mod.rs
@@ -987,7 +987,7 @@ fn populate_allocator(info: &dyn boot::Info, image_address: u64, image_size: u64
 fn init_heap_allocator(size: usize) {
     let (status, heap_start) = ALLOCATOR.borrow_mut().allocate_pages(
         efi::ALLOCATE_ANY_PAGES,
-        efi::BOOT_SERVICES_CODE,
+        efi::RUNTIME_SERVICES_DATA,
         size as u64 / PAGE_SIZE,
         0,
     );