1 files changed, 4 insertions, 1 deletions

diff --git a/litmus/rt_domain.c b/litmus/rt_domain.c
index 62c9fdcd22be..d5ece53fa61e 100644
--- a/litmus/rt_domain.c
+++ b/litmus/rt_domain.c
@@ -19,6 +19,8 @@
 
 #include <litmus/trace.h>
 
+#include <trace/events/sched.h>
+
 #include <litmus/heap.h>
 
 static int dummy_resched(rt_domain_t *rt)
@@ -181,7 +183,8 @@ static void arm_release_timer(rt_domain_t *_rt)
         list_for_each_safe(pos, safe, &list) {
                 /* pick task of work list */
                 t = list_entry(pos, struct task_struct, rt_param.list);
-                sched_trace_task_release(t);
+//              sched_trace_task_release(t);
+                trace_litmus_task_release(t);
                 list_del(pos);
 
                 /* put into release heap while holding release_lock */