diff --git a/drivers/block/nbd.c b/drivers/block/nbd.c
index ffce519bf008..5112e3a03027 100644
--- a/drivers/block/nbd.c
+++ b/drivers/block/nbd.c
@@ -1369,12 +1369,12 @@ static int nbd_genl_reconnect_sock_cb(struct nbd_device *nbd, unsigned long arg)
 		if (!nsock->dead)
 			continue;
 
+		sk_set_memalloc(sock->sk);
 		mutex_lock(&nsock->tx_lock);
 		if (!nsock->dead) {
 			mutex_unlock(&nsock->tx_lock);
 			continue;
 		}
-		sk_set_memalloc(sock->sk);
 		if (nbd->tag_set.timeout)
 			sock->sk->sk_sndtimeo = nbd->tag_set.timeout;
 		atomic_inc(&config->recv_threads);