diff --git a/drivers/misc/mods/mods_irq.c b/drivers/misc/mods/mods_irq.c
index 15b35fd6..4838b94c 100644
--- a/drivers/misc/mods/mods_irq.c
+++ b/drivers/misc/mods/mods_irq.c
@@ -1,5 +1,5 @@
 // SPDX-License-Identifier: GPL-2.0-only
-/* SPDX-FileCopyrightText: Copyright (c) 2008-2023, NVIDIA CORPORATION.  All rights reserved. */
+/* SPDX-FileCopyrightText: Copyright (c) 2008-2024, NVIDIA CORPORATION.  All rights reserved. */
 
 #include "mods_internal.h"
 
@@ -8,11 +8,10 @@
 #include <linux/interrupt.h>
 #include <linux/pci_regs.h>
 #if defined(MODS_HAS_TEGRA) && defined(CONFIG_OF) && defined(CONFIG_OF_IRQ)
-#include <linux/of.h>
 #include <linux/of_irq.h>
 #include <linux/io.h>
 #include <linux/platform_device.h>
-#include <linux/of_device.h>
+#include <linux/of_platform.h>
 #include <linux/gpio.h>
 #include <linux/of_gpio.h>
 #endif
diff --git a/drivers/misc/mods/mods_smmu_drv.c b/drivers/misc/mods/mods_smmu_drv.c
index c7a73fa5..ac062633 100644
--- a/drivers/misc/mods/mods_smmu_drv.c
+++ b/drivers/misc/mods/mods_smmu_drv.c
@@ -1,5 +1,5 @@
 // SPDX-License-Identifier: GPL-2.0-only
-/* SPDX-FileCopyrightText: Copyright (c) 2019-2023, NVIDIA CORPORATION.  All rights reserved. */
+/* SPDX-FileCopyrightText: Copyright (c) 2019-2024, NVIDIA CORPORATION.  All rights reserved. */
 
 #include <linux/kernel.h>
 #include <linux/device.h>
@@ -12,8 +12,8 @@
 #include <linux/version.h>
 #include <linux/dma-buf.h>
 
-#include <linux/of.h>
 #include <linux/of_platform.h>
+#include <linux/platform_device.h>
 #include <linux/acpi.h>
 #include "mods_internal.h"