* hugetlb_pte_count.
                 */
                if (pmd_val(pmd) & _PAGE_PMD_HUGE) {
-                       if (is_huge_zero_page(pmd_page(pmd)))
+                       if (is_huge_zero_pmd(pmd))
                                mm->context.hugetlb_pte_count++;
                        else
                                mm->context.thp_pte_count++;
                } else {
-                       if (is_huge_zero_page(pmd_page(orig)))
+                       if (is_huge_zero_pmd(orig))
                                mm->context.hugetlb_pte_count--;
                        else
                                mm->context.thp_pte_count--;
         * Sanity check pmd before doing the actual decrement.
         */
        if ((pmd_val(entry) & _PAGE_PMD_HUGE) &&
-           !is_huge_zero_page(pmd_page(entry)))
+           !is_huge_zero_pmd(entry))
                (vma->vm_mm)->context.thp_pte_count--;
 
        return old;