--- x/include/linux/skmsg.h
+++ y/include/linux/skmsg.h
@@ -430,6 +430,10 @@ static inline void sk_psock_restore_prot
 {
 	if (psock->psock_update_sk_prot)
 		psock->psock_update_sk_prot(sk, psock, true);
+	else if (sk->sk_prot->psock_update_sk_prot)
+		sk->sk_prot->psock_update_sk_prot(sk, psock, true);
+	else
+		BUG_ON(1);
 }
 
 static inline struct sk_psock *sk_psock_get(struct sock *sk)