1 files changed, 5 insertions, 1 deletions

diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index 9612ed5831a1..5a196e1b0fbd 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -198,11 +198,15 @@ feather_callback void do_sched_trace_task_completion(unsigned long id,
         struct task_struct *t = (struct task_struct*) _task;
         struct st_event_record* rec = get_record(ST_COMPLETION, t);
         if (rec) {
-                rec->data.completion.when   = tsk_rt(t)->user_job.exec_time;
+                rec->data.completion.exec   = tsk_rt(t)->user_job.exec_time;
                 rec->data.completion.flush  = tsk_rt(t)->flush;
                 rec->data.completion.load   = tsk_rt(t)->load;
+                rec->data.completion.flush_work = tsk_rt(t)->flush_work;
+                rec->data.completion.load_work  = tsk_rt(t)->load_work;
                 tsk_rt(t)->flush = 0;
                 tsk_rt(t)->load  = 0;
+                tsk_rt(t)->flush_work = 0;
+                tsk_rt(t)->load_work  = 0;
                 put_record(rec);
         }
 }