1 files changed, 4 insertions, 4 deletions

diff --git a/mm/hugetlb.c b/mm/hugetlb.c
index 51c9e2c01640..ddd141cad77f 100644
--- a/mm/hugetlb.c
+++ b/mm/hugetlb.c
@@ -97,11 +97,11 @@ static struct page *dequeue_huge_page_vma(struct vm_area_struct *vma,
         struct mempolicy *mpol;
         struct zonelist *zonelist = huge_zonelist(vma, address,
                                         htlb_alloc_mask, &mpol);
-        struct zone **z;
+        struct zone *zone, **z;
 
-        for (z = zonelist->zones; *z; z++) {
-                nid = zone_to_nid(*z);
-                if (cpuset_zone_allowed_softwall(*z, htlb_alloc_mask) &&
+        for_each_zone_zonelist(zone, z, zonelist, MAX_NR_ZONES - 1) {
+                nid = zone_to_nid(zone);
+                if (cpuset_zone_allowed_softwall(zone, htlb_alloc_mask) &&
                     !list_empty(&hugepage_freelists[nid])) {
                         page = list_entry(hugepage_freelists[nid].next,
                                           struct page, lru);