1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/ide/ide-xfer-mode.c b/drivers/ide/ide-xfer-mode.c
index 46d203ce60cc..cdae463f6b41 100644
--- a/drivers/ide/ide-xfer-mode.c
+++ b/drivers/ide/ide-xfer-mode.c
@@ -135,6 +135,7 @@ int ide_set_pio_mode(ide_drive_t *drive, const u8 mode)
          * set transfer mode on the device in ->set_pio_mode method...
          */
         if (port_ops->set_dma_mode == NULL) {
+                drive->pio_mode = mode;
                 port_ops->set_pio_mode(drive, mode - XFER_PIO_0);
                 return 0;
         }
@@ -142,9 +143,11 @@ int ide_set_pio_mode(ide_drive_t *drive, const u8 mode)
         if (hwif->host_flags & IDE_HFLAG_POST_SET_MODE) {
                 if (ide_config_drive_speed(drive, mode))
                         return -1;
+                drive->pio_mode = mode;
                 port_ops->set_pio_mode(drive, mode - XFER_PIO_0);
                 return 0;
         } else {
+                drive->pio_mode = mode;
                 port_ops->set_pio_mode(drive, mode - XFER_PIO_0);
                 return ide_config_drive_speed(drive, mode);
         }