diff --git a/arch/riscv/mm/init.c b/arch/riscv/mm/init.c
index ca03762a373382e04af23bfbf5417c09acb06356..f750e012dbe588e03cb2eb549422912c4d7fbd83 100644
--- a/arch/riscv/mm/init.c
+++ b/arch/riscv/mm/init.c
@@ -515,6 +515,7 @@ asmlinkage void __init setup_vm(uintptr_t dtb_pa)
 #else
 	dtb_early_va = (void *)dtb_pa;
 #endif
+	dtb_early_pa = dtb_pa;
 }
 
 static inline void setup_vm_final(void)