diff --git a/uefiapi/target/msm8226.c b/uefiapi/target/msm8226.c
index 47915ded8..3c21b4b5b 100644
--- a/uefiapi/target/msm8226.c
+++ b/uefiapi/target/msm8226.c
@@ -1,4 +1,5 @@
 #include <err.h>
+#include <board.h>
 #include <stdint.h>
 #include <pm8x41.h>
 #include <dev/newkeys.h>
@@ -43,8 +44,8 @@ void uefiapi_platform_init_post(void)
 void* uefiapi_mmap_add_platform_mappings(void *pdata, lkapi_mmap_add_cb_t cb)
 {
     // iomap
-    pdata = cb(pdata, MSM_IOMAP_BASE, (MSM_IOMAP_END - MSM_IOMAP_BASE), LKAPI_MMAP_RANGEFLAG_RESERVED,
-                      LKAPI_MEMORYATTR_DEVICE, 0, LKAPI_MMAP_RANGEFLAG_UNUSED|LKAPI_MMAP_RANGEFLAG_DRAM);
+    pdata = cb(pdata, MSM_IOMAP_BASE, 1*MB, LKAPI_MMAP_RANGEFLAG_RESERVED,
+               LKAPI_MEMORYATTR_DEVICE, 0, LKAPI_MMAP_RANGEFLAG_UNUSED|LKAPI_MMAP_RANGEFLAG_DRAM);
 
     return pdata;
 }