1 files changed, 2 insertions, 2 deletions

diff --git a/mm/memory.c b/mm/memory.c
index ab650c21bccd..1aa5c03566f1 100644
--- a/mm/memory.c
+++ b/mm/memory.c
@@ -356,7 +356,7 @@ void free_pgd_range(struct mmu_gather *tlb,
          * We add page table cache pages with PAGE_SIZE,
          * (see pte_free_tlb()), flush the tlb if we need
          */
-        tlb_remove_check_page_size_change(tlb, PAGE_SIZE);
+        tlb_change_page_size(tlb, PAGE_SIZE);
         pgd = pgd_offset(tlb->mm, addr);
         do {
                 next = pgd_addr_end(addr, end);
@@ -1046,7 +1046,7 @@ static unsigned long zap_pte_range(struct mmu_gather *tlb,
         pte_t *pte;
         swp_entry_t entry;
 
-        tlb_remove_check_page_size_change(tlb, PAGE_SIZE);
+        tlb_change_page_size(tlb, PAGE_SIZE);
 again:
         init_rss_vec(rss);
         start_pte = pte_offset_map_lock(mm, pmd, addr, &ptl);