1 files changed, 1 insertions, 1 deletions

diff --git a/mm/page_alloc.c b/mm/page_alloc.c
index c4cdd3954804..2c74367a8eba 100644
--- a/mm/page_alloc.c
+++ b/mm/page_alloc.c
@@ -5926,7 +5926,7 @@ void __ref memmap_init_zone_device(struct zone *zone,
         unsigned long start = jiffies;
         int nid = pgdat->node_id;
 
-        if (WARN_ON_ONCE(!pgmap || !is_dev_zone(zone)))
+        if (WARN_ON_ONCE(!pgmap || zone_idx(zone) != ZONE_DEVICE))
                 return;
 
         /*