1 files changed, 2 insertions, 0 deletions

diff --git a/include/linux/skmsg.h b/include/linux/skmsg.h
index 178a3933a71b..50ced8aba9db 100644
--- a/include/linux/skmsg.h
+++ b/include/linux/skmsg.h
@@ -351,6 +351,8 @@ static inline void sk_psock_update_proto(struct sock *sk,
 static inline void sk_psock_restore_proto(struct sock *sk,
                                           struct sk_psock *psock)
 {
+        sk->sk_write_space = psock->saved_write_space;
+
         if (psock->sk_proto) {
                 sk->sk_prot = psock->sk_proto;
                 psock->sk_proto = NULL;