2 files changed, 4 insertions, 4 deletions

diff --git a/net/core/scm.c b/net/core/scm.c
index 83b2b383c865..c77b7c3d2f9d 100644
--- a/net/core/scm.c
+++ b/net/core/scm.c
@@ -291,7 +291,7 @@ void scm_detach_fds(struct msghdr *msg, struct scm_cookie *scm)
                 sock = sock_from_file(fp[i], &err);
                 if (sock) {
                         sock_update_netprioidx(sock->sk, current);
-                        sock_update_classid(sock->sk, current);
+                        sock_update_classid(sock->sk);
                 }
                 fd_install(new_fd, get_file(fp[i]));
         }
diff --git a/net/core/sock.c b/net/core/sock.c
index 2ff5f3619a8d..0e1d2fe827ec 100644
--- a/net/core/sock.c
+++ b/net/core/sock.c
@@ -1307,11 +1307,11 @@ static void sk_prot_free(struct proto *prot, struct sock *sk)
 }
 
 #if IS_ENABLED(CONFIG_NET_CLS_CGROUP)
-void sock_update_classid(struct sock *sk, struct task_struct *task)
+void sock_update_classid(struct sock *sk)
 {
         u32 classid;
 
-        classid = task_cls_classid(task);
+        classid = task_cls_classid(current);
         if (classid != sk->sk_classid)
                 sk->sk_classid = classid;
 }
@@ -1353,7 +1353,7 @@ struct sock *sk_alloc(struct net *net, int family, gfp_t priority,
                 sock_net_set(sk, get_net(net));
                 atomic_set(&sk->sk_wmem_alloc, 1);
 
-                sock_update_classid(sk, current);
+                sock_update_classid(sk);
                 sock_update_netprioidx(sk, current);
         }