diff --git a/kernel/conftest.sh b/kernel/conftest.sh
index 83700ad..5a0ce4d 100755
--- a/kernel/conftest.sh
+++ b/kernel/conftest.sh
@@ -177,6 +177,7 @@ test_headers() {
     FILES="$FILES linux/file.h"
 
     FILES_ARCH="$FILES_ARCH asm/set_memory.h"
+    FILES_ARCH="$FILES_ARCH asm/pgtable.h"
 
     translate_and_find_header_files $HEADERS      $FILES
     translate_and_find_header_files $HEADERS_ARCH $FILES_ARCH
@@ -440,6 +441,9 @@ compile_test() {
             # Determine if the set_memory_array_uc() function is present.
             #
             CODE="
+            #if defined(NV_ASM_PGTABLE_H_PRESENT)
+            #include <asm/pgtable.h>
+            #endif
             #if defined(NV_ASM_SET_MEMORY_H_PRESENT)
             #include <asm/set_memory.h>
             #else