diff --git a/bootctrl/Android.bp b/bootctrl/Android.bp
index a84200e..d53baa6 100644
--- a/bootctrl/Android.bp
+++ b/bootctrl/Android.bp
@@ -27,7 +27,7 @@
     srcs: ["BootControl.cpp", "boot_region_control.cpp"],
     cppflags: ["-Wno-unused-variable"],
     cflags: ["-Wno-unused-variable"],
-    header_libs: ["device_kernel_headers"],
+    header_libs: ["generated_kernel_headers"],
 
     shared_libs: [
         "liblog",
diff --git a/libmtk_bsg/Android.bp b/libmtk_bsg/Android.bp
index 8665c5b..7156283 100644
--- a/libmtk_bsg/Android.bp
+++ b/libmtk_bsg/Android.bp
@@ -10,6 +10,6 @@
     recovery_available: true,
     srcs: ["*.c"],
 
-    header_libs: ["device_kernel_headers"],
+    header_libs: ["generated_kernel_headers"],
     export_include_dirs: ["include"],
 }