1 files changed, 13 insertions, 2 deletions

diff --git a/litmus/trace.c b/litmus/trace.c
index 49f8246cda50..1d4593aec71d 100644
--- a/litmus/trace.c
+++ b/litmus/trace.c
@@ -5,6 +5,10 @@
 #include <litmus/litmus.h>
 #include <litmus/trace.h>
 
+#include <litmus/domain.h>
+#include <litmus/event_group.h>
+#include <litmus/sched_mc.h>
+
 /******************************************************************************/
 /*                          Allocation                                        */
 /******************************************************************************/
@@ -81,8 +85,15 @@ feather_callback void save_timestamp_def(unsigned long event,
 feather_callback void save_timestamp_task(unsigned long event,
                                           unsigned long t_ptr)
 {
-        int rt = is_realtime((struct task_struct *) t_ptr);
-        __save_timestamp(event, rt ? TSK_RT : TSK_BE);
+        struct task_struct *ts = (struct task_struct*) t_ptr;
+        int rt = is_realtime(ts);
+        uint8_t type = rt ? TSK_RT : TSK_BE;
+
+        if (TS_LVLA_SCHED_END_ID == event) {
+                if (rt && CRIT_LEVEL_A == tsk_mc_crit(ts))
+                        type = TSK_LVLA;
+        }
+        __save_timestamp(event, type);
 }
 
 feather_callback void save_timestamp_cpu(unsigned long event,