1 files changed, 4 insertions, 4 deletions

diff --git a/arch/powerpc/kernel/idle.c b/arch/powerpc/kernel/idle.c
index 0a48bf5db6c8..65035141552b 100644
--- a/arch/powerpc/kernel/idle.c
+++ b/arch/powerpc/kernel/idle.c
@@ -101,11 +101,11 @@ void cpu_idle(void)
                 ppc64_runlatch_on();
                 rcu_idle_exit();
                 tick_nohz_idle_exit();
-                preempt_enable_no_resched();
-                if (cpu_should_die())
+                if (cpu_should_die()) {
+                        preempt_enable_no_resched();
                         cpu_die();
-                schedule();
-                preempt_disable();
+                }
+                schedule_preempt_disabled();
         }
 }