*/
 struct scsi_cmnd *scsi_get_command(struct scsi_device *dev, gfp_t gfp_mask)
 {
-       struct scsi_cmnd *cmd;
+       struct scsi_cmnd *cmd = __scsi_get_command(dev->host, gfp_mask);
+       unsigned long flags;
 
-       /* Bail if we can't get a reference to the device */
-       if (!get_device(&dev->sdev_gendev))
+       if (unlikely(cmd == NULL))
                return NULL;
 
-       cmd = __scsi_get_command(dev->host, gfp_mask);
-
-       if (likely(cmd != NULL)) {
-               unsigned long flags;
-
-               cmd->device = dev;
-               INIT_LIST_HEAD(&cmd->list);
-               INIT_DELAYED_WORK(&cmd->abort_work, scmd_eh_abort_handler);
-               spin_lock_irqsave(&dev->list_lock, flags);
-               list_add_tail(&cmd->list, &dev->cmd_list);
-               spin_unlock_irqrestore(&dev->list_lock, flags);
-               cmd->jiffies_at_alloc = jiffies;
-       } else
-               put_device(&dev->sdev_gendev);
-
+       cmd->device = dev;
+       INIT_LIST_HEAD(&cmd->list);
+       INIT_DELAYED_WORK(&cmd->abort_work, scmd_eh_abort_handler);
+       spin_lock_irqsave(&dev->list_lock, flags);
+       list_add_tail(&cmd->list, &dev->cmd_list);
+       spin_unlock_irqrestore(&dev->list_lock, flags);
+       cmd->jiffies_at_alloc = jiffies;
        return cmd;
 }
 EXPORT_SYMBOL(scsi_get_command);
  * __scsi_put_command - Free a struct scsi_cmnd
  * @shost: dev->host
  * @cmd: Command to free
- * @dev: parent scsi device
  */
-void __scsi_put_command(struct Scsi_Host *shost, struct scsi_cmnd *cmd,
-                       struct device *dev)
+void __scsi_put_command(struct Scsi_Host *shost, struct scsi_cmnd *cmd)
 {
        unsigned long flags;
 
 
        if (likely(cmd != NULL))
                scsi_pool_free_command(shost->cmd_pool, cmd);
-
-       put_device(dev);
 }
 EXPORT_SYMBOL(__scsi_put_command);
 
  */
 void scsi_put_command(struct scsi_cmnd *cmd)
 {
-       struct scsi_device *sdev = cmd->device;
        unsigned long flags;
 
        /* serious error if the command hasn't come from a device list */
 
        cancel_delayed_work(&cmd->abort_work);
 
-       __scsi_put_command(cmd->device->host, cmd, &sdev->sdev_gendev);
+       __scsi_put_command(cmd->device->host, cmd);
 }
 EXPORT_SYMBOL(scsi_put_command);
 
 
        req->special = NULL;
 
        scsi_put_command(cmd);
+       put_device(&cmd->device->sdev_gendev);
 }
 
 /**
        get_device(&sdev->sdev_gendev);
 
        scsi_put_command(cmd);
+       put_device(&sdev->sdev_gendev);
        scsi_run_queue(q);
 
        /* ok to remove device now */
        scsi_release_buffers(cmd);
        cmd->request->special = NULL;
        scsi_put_command(cmd);
+       put_device(&cmd->device->sdev_gendev);
        return error;
 }
 EXPORT_SYMBOL(scsi_init_io);
        struct scsi_cmnd *cmd;
 
        if (!req->special) {
+               /* Bail if we can't get a reference to the device */
+               if (!get_device(&sdev->sdev_gendev))
+                       return NULL;
+
                cmd = scsi_get_command(sdev, GFP_ATOMIC);
-               if (unlikely(!cmd))
+               if (unlikely(!cmd)) {
+                       put_device(&sdev->sdev_gendev);
                        return NULL;
+               }
                req->special = cmd;
        } else {
                cmd = req->special;
                        struct scsi_cmnd *cmd = req->special;
                        scsi_release_buffers(cmd);
                        scsi_put_command(cmd);
+                       put_device(&cmd->device->sdev_gendev);
                        req->special = NULL;
                }
                break;
 
 extern struct scsi_cmnd *scsi_get_command(struct scsi_device *, gfp_t);
 extern struct scsi_cmnd *__scsi_get_command(struct Scsi_Host *, gfp_t);
 extern void scsi_put_command(struct scsi_cmnd *);
-extern void __scsi_put_command(struct Scsi_Host *, struct scsi_cmnd *,
-                              struct device *);
+extern void __scsi_put_command(struct Scsi_Host *, struct scsi_cmnd *);
 extern void scsi_finish_command(struct scsi_cmnd *cmd);
 
 extern void *scsi_kmap_atomic_sg(struct scatterlist *sg, int sg_count,