1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/ide/ide-devsets.c b/drivers/ide/ide-devsets.c
index 1099bf7cf96..c6935c78757 100644
--- a/drivers/ide/ide-devsets.c
+++ b/drivers/ide/ide-devsets.c
@@ -105,15 +105,17 @@ static int set_pio_mode(ide_drive_t *drive, int arg)
                 return -ENOSYS;
 
         if (set_pio_mode_abuse(drive->hwif, arg)) {
+                drive->pio_mode = arg + XFER_PIO_0;
+
                 if (arg == 8 || arg == 9) {
                         unsigned long flags;
 
                         /* take lock for IDE_DFLAG_[NO_]UNMASK/[NO_]IO_32BIT */
                         spin_lock_irqsave(&hwif->lock, flags);
-                        port_ops->set_pio_mode(drive, arg);
+                        port_ops->set_pio_mode(hwif, drive);
                         spin_unlock_irqrestore(&hwif->lock, flags);
                 } else
-                        port_ops->set_pio_mode(drive, arg);
+                        port_ops->set_pio_mode(hwif, drive);
         } else {
                 int keep_dma = !!(drive->dev_flags & IDE_DFLAG_USING_DMA);