1 files changed, 3 insertions, 2 deletions

diff --git a/mm/page_isolation.c b/mm/page_isolation.c
index 3ca1716471bc..345643b85bd4 100644
--- a/mm/page_isolation.c
+++ b/mm/page_isolation.c
@@ -77,11 +77,12 @@ int set_migratetype_isolate(struct page *page)
 out:
         if (!ret) {
                 unsigned long nr_pages;
+                int migratetype = get_pageblock_migratetype(page);
 
                 set_pageblock_isolate(page);
                 nr_pages = move_freepages_block(zone, page, MIGRATE_ISOLATE);
 
-                __mod_zone_page_state(zone, NR_FREE_PAGES, -nr_pages);
+                __mod_zone_freepage_state(zone, -nr_pages, migratetype);
         }
 
         spin_unlock_irqrestore(&zone->lock, flags);
@@ -100,7 +101,7 @@ void unset_migratetype_isolate(struct page *page, unsigned migratetype)
         if (get_pageblock_migratetype(page) != MIGRATE_ISOLATE)
                 goto out;
         nr_pages = move_freepages_block(zone, page, migratetype);
-        __mod_zone_page_state(zone, NR_FREE_PAGES, nr_pages);
+        __mod_zone_freepage_state(zone, nr_pages, migratetype);
         restore_pageblock_isolate(page, migratetype);
 out:
         spin_unlock_irqrestore(&zone->lock, flags);