diff --git a/recipes-kernel/nvidia-kernel-oot/nvidia-kernel-oot/0001-Makefile-update-for-OE-builds.patch b/recipes-kernel/nvidia-kernel-oot/nvidia-kernel-oot/0001-Makefile-update-for-OE-builds.patch
index faaff34a..0de15118 100644
--- a/recipes-kernel/nvidia-kernel-oot/nvidia-kernel-oot/0001-Makefile-update-for-OE-builds.patch
+++ b/recipes-kernel/nvidia-kernel-oot/nvidia-kernel-oot/0001-Makefile-update-for-OE-builds.patch
@@ -89,7 +89,7 @@ index ad7dc07..64bec51 100644
 -	$(MAKE) -j $(NPROC) ARCH=arm64 \
 -		-C $(KERNEL_OUTPUT) \
 +	$(MAKE) ARCH=arm64 \
-+		-C $(KERNEL_SRC) \
++		-C $(KERNEL_PATH) \
  		M=$(MAKEFILE_DIR)/hwpm/drivers/tegra/hwpm \
 +		MODLIB=$(MODLIB) \
  		CONFIG_TEGRA_OOT_MODULE=m \
@@ -102,7 +102,7 @@ index ad7dc07..64bec51 100644
 -	$(MAKE) -j $(NPROC) ARCH=arm64 \
 -		-C $(KERNEL_OUTPUT) \
 +	$(MAKE) ARCH=arm64 \
-+		-C $(KERNEL_SRC) \
++		-C $(KERNEL_PATH) \
  		M=$(MAKEFILE_DIR)/nvidia-oot \
 +		MODLIB=$(MODLIB) \
  		CONFIG_TEGRA_OOT_MODULE=m \
@@ -117,7 +117,7 @@ index ad7dc07..64bec51 100644
 -	$(MAKE) -j $(NPROC) ARCH=arm64 \
 -		-C $(KERNEL_OUTPUT) \
 +	$(MAKE) ARCH=arm64 \
-+		-C $(KERNEL_SRC) \
++		-C $(KERNEL_PATH) \
  		M=$(MAKEFILE_DIR)/nvgpu/drivers/gpu/nvgpu \
 +		MODLIB=$(MODLIB) \
  		CONFIG_TEGRA_OOT_MODULE=m \
@@ -177,7 +177,7 @@ index ad7dc07..64bec51 100644
  	@echo   "make $(MAKECMDGOALS) - nvidia-display ..."
  	@echo   "================================================================================"
 -	$(MAKE) -C $(NVIDIA_HEADERS) \
-+	$(MAKE) -C $(KERNEL_SRC) \
++	$(MAKE) -C $(KERNEL_PATH) \
 +		KERNEL_MODLIB=$(MODLIB) \
  		M=$(MAKEFILE_DIR)/nvdisplay/kernel-open modules_install
  
diff --git a/recipes-kernel/nvidia-kernel-oot/nvidia-kernel-oot_36.4.0.bb b/recipes-kernel/nvidia-kernel-oot/nvidia-kernel-oot_36.4.0.bb
index 5e71afd7..76393b5e 100644
--- a/recipes-kernel/nvidia-kernel-oot/nvidia-kernel-oot_36.4.0.bb
+++ b/recipes-kernel/nvidia-kernel-oot/nvidia-kernel-oot_36.4.0.bb
@@ -62,7 +62,7 @@ TEGRA_OOT_REPLACEMENT_DRIVERS = "\
 "
 
 EXTRA_OEMAKE += '\
-    IGNORE_PREEMPT_RT_PRESENCE=1 \
+    IGNORE_PREEMPT_RT_PRESENCE=1 KERNEL_PATH="${STAGING_KERNEL_DIR}" \
     CC="${KERNEL_CC}" CXX="${KERNEL_CC} -x c++" LD="${KERNEL_LD}" AR="${KERNEL_AR}" \
     OBJCOPY="${KERNEL_OBJCOPY}" \
 '