diff --git a/arch/risc-v/src/bl808/bl808_mm_init.c b/arch/risc-v/src/bl808/bl808_mm_init.c
index 0d17982a8e3..9245af153fe 100644
--- a/arch/risc-v/src/bl808/bl808_mm_init.c
+++ b/arch/risc-v/src/bl808/bl808_mm_init.c
@@ -307,8 +307,9 @@ void bl808_mm_init(void)
  *
  ****************************************************************************/
 
-void weak_function mmu_flush_cache(void)
+void weak_function mmu_flush_cache(uintptr_t reg)
 {
+  UNUSED(reg);
   __asm__ __volatile__
     (
 
diff --git a/arch/risc-v/src/common/riscv_mmu.h b/arch/risc-v/src/common/riscv_mmu.h
index 23ce40b5258..a2cd9f0ee23 100644
--- a/arch/risc-v/src/common/riscv_mmu.h
+++ b/arch/risc-v/src/common/riscv_mmu.h
@@ -158,7 +158,7 @@ extern uintptr_t g_kernel_pgt_pbase;
  * Public Function Prototypes
  ****************************************************************************/
 
-void weak_function mmu_flush_cache(void);
+void weak_function mmu_flush_cache(uintptr_t);
 
 /****************************************************************************
  * Inline Functions
@@ -216,7 +216,7 @@ static inline void mmu_write_satp(uintptr_t reg)
 
   if (mmu_flush_cache != NULL)
     {
-      mmu_flush_cache();
+      mmu_flush_cache(reg);
     }
 }