diff --git a/mm/page_isolation.c b/mm/page_isolation.c
index f2452e5116b2..72f5ac381ab3 100644
--- a/mm/page_isolation.c
+++ b/mm/page_isolation.c
@@ -68,7 +68,7 @@ int set_migratetype_isolate(struct page *page, bool skip_hwpoisoned_pages)
 
 	spin_unlock_irqrestore(&zone->lock, flags);
 	if (!ret)
-		drain_all_pages(NULL);
+		drain_all_pages(zone);
 	return ret;
 }