1 files changed, 6 insertions, 2 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index de751a14d77c..c4c74fe299fc 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -19,6 +19,8 @@
 
 #include <litmus/trace.h>
 
+#include <trace/events/sched.h>
+
 #include <litmus/rt_domain.h>
 
 /* Number of RT tasks that exist in the system */
@@ -342,7 +344,8 @@ long litmus_admit_task(struct task_struct* tsk)
 
         if (!retval) {
                 sched_trace_task_name(tsk);
-                sched_trace_task_param(tsk);
+//              sched_trace_task_param(tsk);
+                trace_litmus_task_param(tsk);
                 atomic_inc(&rt_task_count);
         }
 
@@ -355,7 +358,8 @@ out:
 void litmus_exit_task(struct task_struct* tsk)
 {
         if (is_realtime(tsk)) {
-                sched_trace_task_completion(tsk, 1);
+//              sched_trace_task_completion(tsk, 1);
+                trace_litmus_task_completion(tsk, 1);
 
                 litmus->task_exit(tsk);