fix level-a schedule event id and a bad define in rt_domain.h

1 files changed, 5 insertions, 2 deletions

diff --git a/litmus/trace.c b/litmus/trace.c
index 417422d2f16d..1d4593aec71d 100644
--- a/litmus/trace.c
+++ b/litmus/trace.c
@@ -4,6 +4,9 @@
 #include <litmus/ftdev.h>
 #include <litmus/litmus.h>
 #include <litmus/trace.h>
+
+#include <litmus/domain.h>
+#include <litmus/event_group.h>
 #include <litmus/sched_mc.h>
 
 /******************************************************************************/
@@ -82,11 +85,11 @@ feather_callback void save_timestamp_def(unsigned long event,
 feather_callback void save_timestamp_task(unsigned long event,
                                           unsigned long t_ptr)
 {
-        struct task_struct *ts = t_ptr;
+        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 == event) {
+        if (TS_LVLA_SCHED_END_ID == event) {
                 if (rt && CRIT_LEVEL_A == tsk_mc_crit(ts))
                         type = TSK_LVLA;
         }