diff --git a/qemu_aarch64/plat.c b/qemu_aarch64/plat.c
index 2ae2686..b9eb98f 100644
--- a/qemu_aarch64/plat.c
+++ b/qemu_aarch64/plat.c
@@ -26,13 +26,13 @@
 	return pa_init(plat_fdt_addr);
 }
 
-void plat_get_initrd_range(paddr_t *begin, paddr_t *end)
+void plat_get_initrd_range(paddr_t *begin, paddr_t *end, struct mpool *ppool)
 {
 	struct fdt_header *fdt;
 	struct fdt_node n;
 
 	/* Get the memory map from the FDT. */
-	fdt = fdt_map(plat_get_fdt_addr(), &n);
+	fdt = fdt_map(plat_get_fdt_addr(), &n, ppool);
 	if (!fdt) {
 		return;
 	}
@@ -45,7 +45,7 @@
 	fdt_find_initrd(&n, begin, end);
 
 out_unmap_fdt:
-	if (!fdt_unmap(fdt)) {
+	if (!fdt_unmap(fdt, ppool)) {
 		dlog("Unable to unmap fdt.");
 	}
 }