diff --git a/arch/x86/mm/kmemcheck/kmemcheck.c b/arch/x86/mm/kmemcheck/kmemcheck.c
index a644f248451b..6931e5115bcd 100644
--- a/arch/x86/mm/kmemcheck/kmemcheck.c
+++ b/arch/x86/mm/kmemcheck/kmemcheck.c
@@ -611,7 +611,6 @@ bool kmemcheck_fault(struct pt_regs *regs, unsigned long address,
 	unsigned long error_code)
 {
 	pte_t *pte;
-	unsigned int level;
 
 	/*
 	 * XXX: Is it safe to assume that memory accesses from virtual 86
@@ -624,13 +623,9 @@ bool kmemcheck_fault(struct pt_regs *regs, unsigned long address,
 	if (regs->cs != __KERNEL_CS)
 		return false;
 
-	pte = lookup_address(address, &level);
+	pte = kmemcheck_pte_lookup(address);
 	if (!pte)
 		return false;
-	if (level != PG_LEVEL_4K)
-		return false;
-	if (!pte_hidden(*pte))
-		return false;
 
 	if (error_code & 2)
 		kmemcheck_access(regs, address, KMEMCHECK_WRITE);