diff --git a/arch/riscv/mm/init.c b/arch/riscv/mm/init.c
index 83f7d12042fbc781ab91a3a90c70043400f66c81..a1ca6200c31fc5729956dadbed6af70c438647e8 100644
--- a/arch/riscv/mm/init.c
+++ b/arch/riscv/mm/init.c
@@ -458,7 +458,7 @@ void __init paging_init(void)
 	zone_sizes_init();
 }
 
-#ifdef CONFIG_SPARSEMEM
+#ifdef CONFIG_SPARSEMEM_VMEMMAP
 int __meminit vmemmap_populate(unsigned long start, unsigned long end, int node,
 			       struct vmem_altmap *altmap)
 {