1 files changed, 2 insertions, 2 deletions

diff --git a/mm/memory.c b/mm/memory.c
index c00bac66ce9f..563792f4f687 100644
--- a/mm/memory.c
+++ b/mm/memory.c
@@ -681,7 +681,7 @@ static unsigned long zap_pte_range(struct mmu_gather *tlb,
                                         mark_page_accessed(page);
                                 file_rss--;
                         }
-                        page_remove_rmap(page);
+                        page_remove_rmap(page, vma);
                         tlb_remove_page(tlb, page);
                         continue;
                 }
@@ -1586,7 +1586,7 @@ gotten:
         page_table = pte_offset_map_lock(mm, pmd, address, &ptl);
         if (likely(pte_same(*page_table, orig_pte))) {
                 if (old_page) {
-                        page_remove_rmap(old_page);
+                        page_remove_rmap(old_page, vma);
                         if (!PageAnon(old_page)) {
                                 dec_mm_counter(mm, file_rss);
                                 inc_mm_counter(mm, anon_rss);