diff --git a/drivers/misc/mods/Makefile b/drivers/misc/mods/Makefile
index 804a57ab..e4980938 100644
--- a/drivers/misc/mods/Makefile
+++ b/drivers/misc/mods/Makefile
@@ -1,6 +1,10 @@
 # SPDX-License-Identifier: GPL-2.0-only
 # Copyright (c) 2017-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 
+ifeq ($(origin srctree.nvidia), undefined)
+srctree.nvidia := $(srctree.nvidia-oot)
+endif
+
 ccflags-y += -I$(srctree.nvidia)/include
 ccflags-y += -DMODS_HAS_TEGRA