1 files changed, 3 insertions, 3 deletions

diff --git a/arch/arm/mach-omap2/cm.c b/arch/arm/mach-omap2/cm.c
index 2d83565d2be2..721c3b66740a 100644
--- a/arch/arm/mach-omap2/cm.c
+++ b/arch/arm/mach-omap2/cm.c
@@ -50,15 +50,15 @@ int omap2_cm_wait_module_ready(s16 prcm_mod, u8 idlest_id, u8 idlest_shift)
 
         cm_idlest_reg = cm_idlest_offs[idlest_id - 1];
 
+        mask = 1 << idlest_shift;
+
         if (cpu_is_omap24xx())
-                ena = idlest_shift;
+                ena = mask;
         else if (cpu_is_omap34xx())
                 ena = 0;
         else
                 BUG();
 
-        mask = 1 << idlest_shift;
-
         /* XXX should be OMAP2 CM */
         omap_test_timeout(((cm_read_mod_reg(prcm_mod, cm_idlest_reg) & mask) == ena),
                           MAX_MODULE_READY_TIME, i);