struct gendisk *disk;
        int blksize;
        loff_t bytesize;
-       pid_t pid; /* pid of nbd-client, if attached */
        int xmit_timeout;
        int disconnect; /* a disconnect has been requested by user */
 
                        struct device_attribute *attr, char *buf)
 {
        struct gendisk *disk = dev_to_disk(dev);
+       struct nbd_device *nbd = (struct nbd_device *)disk->private_data;
 
-       return sprintf(buf, "%ld\n",
-               (long) ((struct nbd_device *)disk->private_data)->pid);
+       return sprintf(buf, "%d\n", task_pid_nr(nbd->task_recv));
 }
 
 static struct device_attribute pid_attr = {
        BUG_ON(nbd->magic != NBD_MAGIC);
 
        sk_set_memalloc(nbd->sock->sk);
-       nbd->pid = task_pid_nr(current);
+
+       nbd->task_recv = current;
+
        ret = device_create_file(disk_to_dev(nbd->disk), &pid_attr);
        if (ret) {
                dev_err(disk_to_dev(nbd->disk), "device_create_file failed!\n");
-               nbd->pid = 0;
+               nbd->task_recv = NULL;
                return ret;
        }
 
-       nbd->task_recv = current;
-
        while (1) {
                req = nbd_read_stat(nbd);
                if (IS_ERR(req)) {
                nbd_end_request(nbd, req);
        }
 
+       device_remove_file(disk_to_dev(nbd->disk), &pid_attr);
+
        nbd->task_recv = NULL;
 
        if (signal_pending(current)) {
                ret = -ETIMEDOUT;
        }
 
-       device_remove_file(disk_to_dev(nbd->disk), &pid_attr);
-       nbd->pid = 0;
        return ret;
 }
 
                struct socket *sock;
                int error;
 
-               if (nbd->pid)
+               if (nbd->task_recv)
                        return -EBUSY;
                if (!nbd->sock)
                        return -EINVAL;