{
        struct vm_area_struct *vma = vmf->vma;
        gfp_t gfp;
-       struct page *page;
+       struct folio *folio;
        unsigned long haddr = vmf->address & HPAGE_PMD_MASK;
 
        if (!transhuge_vma_suitable(vma, haddr))
                return ret;
        }
        gfp = vma_thp_gfp_mask(vma);
-       page = alloc_hugepage_vma(gfp, vma, haddr, HPAGE_PMD_ORDER);
-       if (unlikely(!page)) {
+       folio = vma_alloc_folio(gfp, HPAGE_PMD_ORDER, vma, haddr, true);
+       if (unlikely(!folio)) {
                count_vm_event(THP_FAULT_FALLBACK);
                return VM_FAULT_FALLBACK;
        }
-       prep_transhuge_page(page);
-       return __do_huge_pmd_anonymous_page(vmf, page, gfp);
+       return __do_huge_pmd_anonymous_page(vmf, &folio->page, gfp);
 }
 
 static void insert_pfn_pmd(struct vm_area_struct *vma, unsigned long addr,