diff --git a/mm/page_alloc.c b/mm/page_alloc.c
index cdd8ff4350ba700aa8892f1f976cfe3e4e84707c..d2012e07e529580620b376e19ad24af270679b99 100644
--- a/mm/page_alloc.c
+++ b/mm/page_alloc.c
@@ -1688,12 +1688,13 @@ static int __init deferred_init_memmap(void *data)
 		nr_free += atomic64_read(&args.nr_pages);
 	}
 
-	/* Sanity check that the next zone really is unpopulated */
-	WARN_ON(++zid < MAX_NR_ZONES && populated_zone(++zone));
 	VM_BUG_ON(nr_init != nr_free);
 
 	zone->managed_pages += nr_free;
 
+	/* Sanity check that the next zone really is unpopulated */
+	WARN_ON(++zid < MAX_NR_ZONES && populated_zone(++zone));
+
 	pr_info("node %d initialised, %lu pages in %ums\n", nid, nr_free,
 					jiffies_to_msecs(jiffies - start));