diff --git a/BUILD.bazel b/BUILD.bazel
index 9e58a8cf..d5096cd0 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -152,6 +152,7 @@ kernel_module(
         "drivers/video/tegra/dc/bridge/ti_fpdlink_dp_serializer.ko",
         "drivers/video/tegra/host/capture/nvhost-capture.ko",
         "drivers/video/tegra/host/nvcsi/nvhost-nvcsi.ko",
+        "drivers/video/tegra/host/nvdla/axi/axi-nvdla.ko",
         "drivers/video/tegra/host/nvdla/nvhost-nvdla.ko",
         "drivers/video/tegra/nvmap/nvmap.ko",
         "drivers/video/tegra/tsec/tsecriscv.ko",
diff --git a/drivers/video/tegra/host/nvdla/axi/Makefile b/drivers/video/tegra/host/nvdla/axi/Makefile
index 63138830..4c4e8f6a 100644
--- a/drivers/video/tegra/host/nvdla/axi/Makefile
+++ b/drivers/video/tegra/host/nvdla/axi/Makefile
@@ -11,6 +11,8 @@ ifndef CONFIG_TEGRA_KLEAF_BUILD
 include $(src)/../Makefile.config.mk
 else
 include $(srctree.nvidia-oot)/drivers/video/tegra/host/nvdla/Makefile.config.mk
+ccflags-y += -I$(srctree.nvidia-oot)/drivers/video/tegra/host/nvdla
+ccflags-y += -I$(srctree.nvidia-oot)/drivers/video/tegra/host/nvdla/port/fw
 endif
 
 #