diff --git a/arch/arm/mach-omap2/io.c b/arch/arm/mach-omap2/io.c
index 9bd07f33eea77cfd0a5d4c54bee0d93fd7d6bb84..3fbb0c0b84a40c61488daa0f0d83d49337dd0646 100644
--- a/arch/arm/mach-omap2/io.c
+++ b/arch/arm/mach-omap2/io.c
@@ -49,7 +49,7 @@
  * default mapping provided here.
  */
 
-#ifdef CONFIG_ARCH_OMAP2
+#if defined(CONFIG_SOC_OMAP2420) || defined(CONFIG_SOC_OMAP2430)
 static struct map_desc omap24xx_io_desc[] __initdata = {
 	{
 		.virtual	= L3_24XX_VIRT,