diff --git a/include/linux/sched.h b/include/linux/sched.h
index 4ab9f32f9238a077e0691804daec99463cb225f1..5a63f2d72af6e1e84035bb91df75bd25b843fb6b 100644
--- a/include/linux/sched.h
+++ b/include/linux/sched.h
@@ -2027,7 +2027,7 @@ static inline int fatal_signal_pending(struct task_struct *p)
 
 static inline int need_resched(void)
 {
-	return unlikely(test_tsk_need_resched(current));
+	return unlikely(test_thread_flag(TIF_NEED_RESCHED));
 }
 
 /*