diff --git a/include/net/sock.h b/include/net/sock.h
index 7db3db112baa5eaa9ce1958c7837f89dbafde6a8..e6f235ebf6c9299eba6cea962808cd12129426f0 100644
--- a/include/net/sock.h
+++ b/include/net/sock.h
@@ -897,6 +897,7 @@ static inline void sock_rps_reset_rxhash(struct sock *sk)
 		if (!__rc) {						\
 			*(__timeo) = schedule_timeout(*(__timeo));	\
 		}							\
+		sched_annotate_sleep();						\
 		lock_sock(__sk);					\
 		__rc = __condition;					\
 		__rc;							\