diff --git a/kernel/sched/fair.c b/kernel/sched/fair.c
index a5ae5d85019b..c7f4a2baa7a3 100755
--- a/kernel/sched/fair.c
+++ b/kernel/sched/fair.c
@@ -6505,7 +6505,7 @@ static inline bool walt_target_ok(int target_cpu, int order_index)
 }
 
 static void walt_get_indicies(struct task_struct *p, int *order_index,
-		int *end_index, int task_boost)
+		int *end_index, int task_boost, bool boosted)
 {
 	int i = 0;
 
@@ -6528,8 +6528,7 @@ static void walt_get_indicies(struct task_struct *p, int *order_index,
 		return;
 	}
 
-	if (task_boost == TASK_BOOST_ON_MID ||
-		task_boost_policy(p) == SCHED_BOOST_ON_BIG ||
+	if (boosted || task_boost_policy(p) == SCHED_BOOST_ON_BIG ||
 		walt_task_skip_min_cpu(p))
 		*order_index = 1;
 
@@ -7023,7 +7022,7 @@ int find_energy_efficient_cpu(struct task_struct *p, int prev_cpu,
 	if (unlikely(!cpu_array))
 		goto eas_not_ready;
 
-	walt_get_indicies(p, &order_index, &end_index, task_boost);
+	walt_get_indicies(p, &order_index, &end_index, task_boost, boosted);
 	start_cpu = cpumask_first(&cpu_array[order_index][0]);
 
 	is_rtg = task_in_related_thread_group(p);