1 files changed, 10 insertions, 6 deletions

diff --git a/drivers/ata/libata-core.c b/drivers/ata/libata-core.c
index f6408ca82062..3c1f8830ac8b 100644
--- a/drivers/ata/libata-core.c
+++ b/drivers/ata/libata-core.c
@@ -5685,18 +5685,22 @@ static void ata_host_release(struct device *gendev, void *res)
         for (i = 0; i < host->n_ports; i++) {
                 struct ata_port *ap = host->ports[i];
 
-                if (!ap)
-                        continue;
-
-                if (ap->ops->port_stop)
+                if (ap && ap->ops->port_stop)
                         ap->ops->port_stop(ap);
-
-                scsi_host_put(ap->scsi_host);
         }
 
         if (host->ops->host_stop)
                 host->ops->host_stop(host);
 
+        for (i = 0; i < host->n_ports; i++) {
+                struct ata_port *ap = host->ports[i];
+
+                if (ap)
+                        scsi_host_put(ap->scsi_host);
+
+                host->ports[i] = NULL;
+        }
+
         dev_set_drvdata(gendev, NULL);
 }