diff --git a/mm/vmscan.c b/mm/vmscan.c
index a0245861934a7ac4ac7de1a821e70a28a08b91a8..a51b3c9f05ba1b41204d4ec1b8ebeaedda90cecf 100644
--- a/mm/vmscan.c
+++ b/mm/vmscan.c
@@ -2344,7 +2344,7 @@ static bool sleeping_prematurely(pg_data_t *pgdat, int order, long remaining,
 		}
 
 		if (!zone_watermark_ok_safe(zone, order, high_wmark_pages(zone),
-							classzone_idx, 0))
+							i, 0))
 			all_zones_ok = false;
 		else
 			balanced += zone->present_pages;