diff --git a/gpt-utils/Android.bp b/gpt-utils/Android.bp
index afea692..346897f 100644
--- a/gpt-utils/Android.bp
+++ b/gpt-utils/Android.bp
@@ -27,6 +27,7 @@ cc_library {
     cflags: [
         "-Wall",
         "-Werror",
+        "-D_GENERIC_KERNEL_HEADERS",
     ],
     srcs: [
         "gpt-utils.cpp",
@@ -34,7 +35,7 @@ cc_library {
     ],
     owner: "qti",
     header_libs: [
-        "generated_kernel_headers",
+        "device_kernel_headers",
     ],
     export_include_dirs: ["."],
 }