diff --git a/device-tree/platform/generic-dts/Makefile b/device-tree/platform/generic-dts/Makefile
index b8b915ae..74cdc9c5 100644
--- a/device-tree/platform/generic-dts/Makefile
+++ b/device-tree/platform/generic-dts/Makefile
@@ -13,7 +13,11 @@ include $(oottree)/scripts/Makefile.lib
 
 DTC_CPP_FLAGS += $(EXTRA_CPP_FLAGS)
 
+ifneq ($(findstring oot,$(NV_BUILD_KERNEL_OPTIONS)),)
 DTC_CPP_FLAGS += -DLINUX_VERSION=515 -DTEGRA_HOST1X_DT_VERSION=2
+else
+DTC_CPP_FLAGS += -DLINUX_VERSION=600 -DTEGRA_HOST1X_DT_VERSION=2
+endif
 
 ifneq ($(TEGRA_TOP),)
 SOURCE_TOP := $(TEGRA_TOP)