1 files changed, 4 insertions, 1 deletions

diff --git a/mm/page_cgroup.c b/mm/page_cgroup.c
index 5ddad0c6daa6..6d757e3a872a 100644
--- a/mm/page_cgroup.c
+++ b/mm/page_cgroup.c
@@ -251,6 +251,9 @@ static int __meminit page_cgroup_callback(struct notifier_block *self,
                                 mn->nr_pages, mn->status_change_nid);
                 break;
         case MEM_CANCEL_ONLINE:
+                offline_page_cgroup(mn->start_pfn,
+                                mn->nr_pages, mn->status_change_nid);
+                break;
         case MEM_GOING_OFFLINE:
                 break;
         case MEM_ONLINE:
@@ -271,7 +274,7 @@ void __init page_cgroup_init(void)
         if (mem_cgroup_disabled())
                 return;
 
-        for_each_node_state(nid, N_HIGH_MEMORY) {
+        for_each_node_state(nid, N_MEMORY) {
                 unsigned long start_pfn, end_pfn;
 
                 start_pfn = node_start_pfn(nid);