1 files changed, 3 insertions, 6 deletions

diff --git a/net/ipv4/inet_diag.c b/net/ipv4/inet_diag.c
index 41e1c3ea8b51..56a964a553d2 100644
--- a/net/ipv4/inet_diag.c
+++ b/net/ipv4/inet_diag.c
@@ -336,12 +336,9 @@ int inet_diag_dump_one_icsk(struct inet_hashinfo *hashinfo, struct sk_buff *in_s
                 err = 0;
 
 out:
-        if (sk) {
-                if (sk->sk_state == TCP_TIME_WAIT)
-                        inet_twsk_put((struct inet_timewait_sock *)sk);
-                else
-                        sock_put(sk);
-        }
+        if (sk)
+                sock_gen_put(sk);
+
 out_nosk:
         return err;
 }