diff --git a/mm/internal.h b/mm/internal.h
index f762a34b0c57..6220a5e6b3c7 100644
--- a/mm/internal.h
+++ b/mm/internal.h
@@ -144,7 +144,7 @@ struct alloc_context {
 	bool spread_dirty_pages;
 };
 
-#define ac_classzone_idx(ac) zonelist_zone_idx(ac->preferred_zoneref)
+#define ac_classzone_idx(ac) (ac->high_zoneidx)
 
 /*
  * Locate the struct page for both the matching buddy in our