diff --git a/mm/memblock.c b/mm/memblock.c
index 5c3503c98b2f113ca3de1f0a76e863a316d5c8cd..b91df5cf54d346c59e856c3dd7fba6dc663ee39f 100644
--- a/mm/memblock.c
+++ b/mm/memblock.c
@@ -1692,7 +1692,7 @@ void __init memblock_cap_memory_range(phys_addr_t base, phys_addr_t size)
 	if (!size)
 		return;
 
-	if (memblock.memory.cnt <= 1) {
+	if (!memblock_memory->total_size) {
 		pr_warn("%s: No memory registered yet\n", __func__);
 		return;
 	}