diff --git a/include/linux/tegra-oot-prod.h b/include/linux/tegra-oot-prod.h
index de0021c5..19b7ec17 100644
--- a/include/linux/tegra-oot-prod.h
+++ b/include/linux/tegra-oot-prod.h
@@ -9,7 +9,7 @@
 #include <linux/version.h>
 
 #if LINUX_VERSION_CODE >= KERNEL_VERSION(5, 16, 0)
-#ifdef CONFIG_TEGRA_PROD_ENABLE
+#ifdef CONFIG_TEGRA_PROD_LEGACY
 #include <soc/tegra/tegra-prod.h>
 #include <linux/tegra_prod.h>
 #else
@@ -17,7 +17,7 @@
 #include <linux/tegra-prod-legacy-dummy.h>
 #endif
 #else /* LINUX_VERSION_CODE */
-#ifdef CONFIG_TEGRA_PROD_ENABLE
+#ifdef CONFIG_TEGRA_PROD_LEGACY
 #include <linux/tegra_prod.h>
 #else
 #include <linux/tegra-prod-legacy-dummy.h>