diff --git a/mm/memcontrol.c b/mm/memcontrol.c
index 80bc12a3d4bbf8650ab40b3ffe6c29a4e7e5a882..b68ab450c8d9b6e9124ffe85f6ef5b75634aad8d 100644
--- a/mm/memcontrol.c
+++ b/mm/memcontrol.c
@@ -7049,6 +7049,7 @@ static void uncharge_page(struct page *page, struct uncharge_gather *ug)
 			uncharge_gather_clear(ug);
 		}
 		ug->memcg = page_memcg(page);
+		ug->dummy_page = page;
 
 		/* pairs with css_put in uncharge_batch */
 		css_get(&ug->memcg->css);
@@ -7062,7 +7063,6 @@ static void uncharge_page(struct page *page, struct uncharge_gather *ug)
 	else
 		ug->pgpgout++;
 
-	ug->dummy_page = page;
 	page->memcg_data = 0;
 	css_put(&ug->memcg->css);
 }