diff --git a/BUILD.bazel b/BUILD.bazel
index e0b6a53..82055a7 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -31,9 +31,9 @@ cc_library(
         [
             "src/**/*.cpp",
         ],
-        exclude = ["src/codegen/**", "src/core/**/*.cpp"],
+        exclude = ["src/codegen/**", "src/core/**"],
     ) + select({
-        ":mlir_codegen": glob(["src/codegen/**/*.cpp", "src/core/**/*.cpp"]),
+        ":mlir_codegen": glob(["src/codegen/**/*.cpp", "src/core/**/*.cpp", , "src/core/**/*.h"]),
         "//conditions:default": [],
     }) + [
         ":src/version.cpp",