enum {
        NTEST                   = 400,
+       LOWEST_LEVEL            = PG_LEVEL_4K,
 #ifdef CONFIG_X86_64
-       LOWEST_LEVEL            = 4,
        LPS                     = (1 << PMD_SHIFT),
 #elif defined(CONFIG_X86_PAE)
-       LOWEST_LEVEL            = 4,
        LPS                     = (1 << PMD_SHIFT),
 #else
-       LOWEST_LEVEL            = 4, /* lookup_address lies here */
        LPS                     = (1 << 22),
 #endif
        GPS                     = (1<<30)
 
        pud_t *pud;
        pmd_t *pmd;
 
+       *level = PG_LEVEL_NONE;
+
        if (pgd_none(*pgd))
                return NULL;
        pud = pud_offset(pgd, address);
        pmd = pmd_offset(pud, address);
        if (pmd_none(*pmd))
                return NULL;
-       *level = 3;
+
+       *level = PG_LEVEL_2M;
        if (pmd_large(*pmd))
                return (pte_t *)pmd;
-       *level = 4;
 
+       *level = PG_LEVEL_4K;
        return pte_offset_kernel(pmd, address);
 }
 
                address < (unsigned long)&_etext &&
               (pgprot_val(prot) & _PAGE_NX));
 
-       if (level == 4) {
+       if (level == PG_LEVEL_4K) {
                set_pte_atomic(kpte, mk_pte(page, canon_pgprot(prot)));
        } else {
                err = split_large_page(kpte, address);