diff --git a/linux_fdt/plat.c b/linux_fdt/plat.c
index 8f03174..7956b5f 100644
--- a/linux_fdt/plat.c
+++ b/linux_fdt/plat.c
@@ -27,13 +27,14 @@
 	return pa_init(plat_fdt_addr);
 }
 
-void plat_get_initrd_range(paddr_t *begin, paddr_t *end, struct mpool *ppool)
+void plat_get_initrd_range(struct mm_stage1_locked stage1_locked,
+			   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, ppool);
+	fdt = fdt_map(stage1_locked, plat_get_fdt_addr(), &n, ppool);
 	if (!fdt) {
 		return;
 	}
@@ -46,7 +47,7 @@
 	fdt_find_initrd(&n, begin, end);
 
 out_unmap_fdt:
-	if (!fdt_unmap(fdt, ppool)) {
+	if (!fdt_unmap(stage1_locked, fdt, ppool)) {
 		dlog("Unable to unmap fdt.");
 	}
 }