diff --git a/kernel/sched/walt/walt_rt.c b/kernel/sched/walt/walt_rt.c
index 18d531791ff2..2f57e9faf0b8 100644
--- a/kernel/sched/walt/walt_rt.c
+++ b/kernel/sched/walt/walt_rt.c
@@ -16,7 +16,8 @@ DEFINE_PER_CPU(u64, rt_task_arrival_time) = 0;
 static bool long_running_rt_task_trace_rgstrd;
 
 static void rt_task_arrival_marker(void *unused, bool preempt,
-	struct task_struct *prev, struct task_struct *next)
+	struct task_struct *prev, struct task_struct *next,
+	unsigned int prev_state)
 {
 	unsigned int cpu = raw_smp_processor_id();
 
diff --git a/kernel/sched/walt/walt_tp.c b/kernel/sched/walt/walt_tp.c
index 62d3acb2b482..8d7e2e4528c5 100644
--- a/kernel/sched/walt/walt_tp.c
+++ b/kernel/sched/walt/walt_tp.c
@@ -50,7 +50,8 @@ static void setup_prev_cnts(u32 cpu, u32 cnten_val)
 }
 
 void tracectr_notifier(void *ignore, bool preempt,
-			struct task_struct *prev, struct task_struct *next)
+			struct task_struct *prev, struct task_struct *next,
+			unsigned int prev_state)
 {
 	u32 cnten_val;
 	int current_pid;