}
 
 #ifdef CONFIG_MMU_NOTIFIER
-       if (mmu_notifier_clear_young(mm, addr,
-                               addr + ((1UL) << HPAGE_PMD_SHIFT)))
+       if (mmu_notifier_clear_young(mm, addr, addr + HPAGE_PMD_SIZE))
                referenced = true;
 #endif /* CONFIG_MMU_NOTIFIER */
 
 
                        result->accessed = pmd_young(*pvmw.pmd) ||
                                !folio_test_idle(folio) ||
                                mmu_notifier_test_young(vma->vm_mm, addr);
-                       result->page_sz = ((1UL) << HPAGE_PMD_SHIFT);
+                       result->page_sz = HPAGE_PMD_SIZE;
 #else
                        WARN_ON_ONCE(1);
 #endif /* CONFIG_TRANSPARENT_HUGEPAGE */
 
                if (pmd_young(*pmd) || !page_is_idle(page) ||
                                        mmu_notifier_test_young(walk->mm,
                                                addr)) {
-                       *priv->page_sz = ((1UL) << HPAGE_PMD_SHIFT);
+                       *priv->page_sz = HPAGE_PMD_SIZE;
                        priv->young = true;
                }
                put_page(page);