1 files changed, 3 insertions, 3 deletions

diff --git a/include/trace/events/sched.h b/include/trace/events/sched.h
index 306b31de5194..bc01e06bc716 100644
--- a/include/trace/events/sched.h
+++ b/include/trace/events/sched.h
@@ -116,9 +116,9 @@ static inline long __trace_sched_switch_state(bool preempt, struct task_struct *
          * RUNNING (we will not have dequeued if state != RUNNING).
          */
         if (preempt)
-                return TASK_STATE_MAX;
+                return TASK_REPORT_MAX;
 
-        return task_state_index(p);
+        return 1 << task_state_index(p);
 }
 #endif /* CREATE_TRACE_POINTS */
 
@@ -164,7 +164,7 @@ TRACE_EVENT(sched_switch,
                                 { 0x40, "P" }, { 0x80, "I" }) :
                   "R",
 
-                __entry->prev_state & TASK_STATE_MAX ? "+" : "",
+                __entry->prev_state & TASK_REPORT_MAX ? "+" : "",
                 __entry->next_comm, __entry->next_pid, __entry->next_prio)
 );