diff --git a/include/trace/hooks/sched.h b/include/trace/hooks/sched.h
index 8a8b5a7bda43..e978b679219f 100644
--- a/include/trace/hooks/sched.h
+++ b/include/trace/hooks/sched.h
@@ -318,8 +318,8 @@ DECLARE_RESTRICTED_HOOK(android_rvh_update_thermal_stats,
 		TP_ARGS(cpu), 1);
 
 DECLARE_HOOK(android_vh_do_wake_up_sync,
-	TP_PROTO(struct wait_queue_head *wq_head, int *done),
-	TP_ARGS(wq_head, done));
+	TP_PROTO(struct wait_queue_head *wq_head, int *done, struct sock *sk),
+	TP_ARGS(wq_head, done, sk));
 
 DECLARE_HOOK(android_vh_set_wake_flags,
 	TP_PROTO(int *wake_flags, unsigned int *mode),
diff --git a/net/core/sock.c b/net/core/sock.c
index 059577f8ece5..c71c2281a7c5 100644
--- a/net/core/sock.c
+++ b/net/core/sock.c
@@ -3276,7 +3276,7 @@ void sock_def_readable(struct sock *sk)
 	if (skwq_has_sleeper(wq)) {
 		int done = 0;
 
-		trace_android_vh_do_wake_up_sync(&wq->wait, &done);
+		trace_android_vh_do_wake_up_sync(&wq->wait, &done, sk);
 		if (done)
 			goto out;