diff --git a/mm/page_alloc.c b/mm/page_alloc.c
index 0888870e3458ddd3bb3fa8240e62c1f816c4dc61..6bf3d4461430e18f55295f24c3d8d15bb0359c02 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));