2 files changed, 4 insertions, 1 deletions

diff --git a/include/litmus/sched_trace.h b/include/litmus/sched_trace.h
index f7b2189374b2..7ca34cb13881 100644
--- a/include/litmus/sched_trace.h
+++ b/include/litmus/sched_trace.h
@@ -23,7 +23,8 @@ struct st_param_data {		/* regular params */
         u32     period;
         u32     phase;
         u8      partition;
-        u8      __unused[3];
+        u8      class;
+        u8      __unused[2];
 };
 
 struct st_release_data {        /* A job is was/is going to be released. */
@@ -40,6 +41,7 @@ struct st_assigned_data {	/* A job was asigned to a CPU. 		 */
 struct st_switch_to_data {      /* A process was switched to on a given CPU.   */
         u64     when;           /* When did this occur?                        */
         u32     exec_time;      /* Time the current job has executed.          */
+        u8      __unused[4];
 
 };
 
diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index bd490054f183..5ef8d09ab41f 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -131,6 +131,7 @@ feather_callback void do_sched_trace_task_param(unsigned long id, unsigned long
                 rec->data.param.period    = get_rt_period(t);
                 rec->data.param.phase     = get_rt_phase(t);
                 rec->data.param.partition = get_partition(t);
+                rec->data.param.class     = get_class(t);
                 put_record(rec);
         }
 }