1 files changed, 5 insertions, 0 deletions

diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index 5ef8d09ab41f..768a2a0f9d2f 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -14,7 +14,11 @@
 #include <litmus/sched_trace.h>
 #include <litmus/feather_trace.h>
 #include <litmus/ftdev.h>
+#include <litmus/rt_domain.h>
+#include <litmus/domain.h>
+#include <litmus/event_group.h>
 
+#include <litmus/sched_mc.h>
 
 #define NO_EVENTS               (1 << CONFIG_SCHED_TASK_TRACE_SHIFT)
 
@@ -132,6 +136,7 @@ feather_callback void do_sched_trace_task_param(unsigned long id, unsigned long
                 rec->data.param.phase     = get_rt_phase(t);
                 rec->data.param.partition = get_partition(t);
                 rec->data.param.class     = get_class(t);
+                rec->data.param.level     = (tsk_mc_data(t) ? tsk_mc_crit(t) : -1);
                 put_record(rec);
         }
 }