1 files changed, 6 insertions, 4 deletions
diff --git a/cpukit/rtems/src/taskwakeafter.c b/cpukit/rtems/src/taskwakeafter.c
index 6f0322723a..b7f328f5bc 100644
--- a/cpukit/rtems/src/taskwakeafter.c
+++ b/cpukit/rtems/src/taskwakeafter.c
@@ -30,23 +30,25 @@ rtems_status_code rtems_task_wake_after(
    * It is critical to obtain the executing thread after thread dispatching is
    * disabled on SMP configurations.
    */
-  Thread_Control *executing;
+  Thread_Control  *executing;
+  Per_CPU_Control *cpu_self;
 
-  _Thread_Disable_dispatch();
+  cpu_self = _Thread_Dispatch_disable();
     executing = _Thread_Executing;
 
     if ( ticks == 0 ) {
       _Thread_Yield( executing );
     } else {
       _Thread_Set_state( executing, STATES_DELAYING );
+      _Thread_Wait_flags_set( executing, THREAD_WAIT_STATE_BLOCKED );
       _Watchdog_Initialize(
         &executing->Timer,
-        _Thread_Delay_ended,
+        _Thread_Timeout,
         0,
         executing
       );
       _Watchdog_Insert_ticks( &executing->Timer, ticks );
     }
-  _Thread_Enable_dispatch();
+  _Thread_Dispatch_enable( cpu_self );
   return RTEMS_SUCCESSFUL;
 }