diff --git a/mm/huge_memory.c b/mm/huge_memory.c
index 7de94775a359..eb913acb03b7 100644
--- a/mm/huge_memory.c
+++ b/mm/huge_memory.c
@@ -3049,6 +3049,8 @@ int zap_huge_pud(struct mmu_gather *tlb, struct vm_area_struct *vma,
 	spinlock_t *ptl;
 	pud_t orig_pud;
 
+	tlb_change_page_size(tlb, HPAGE_PUD_SIZE);
+
 	ptl = __pud_trans_huge_lock(pud, vma);
 	if (!ptl)
 		return 0;