1 files changed, 7 insertions, 1 deletions

diff --git a/drivers/scsi/ufs/ufshcd-pci.c b/drivers/scsi/ufs/ufshcd-pci.c
index afaabe2aeac8..7a6edbc55f92 100644
--- a/drivers/scsi/ufs/ufshcd-pci.c
+++ b/drivers/scsi/ufs/ufshcd-pci.c
@@ -164,7 +164,13 @@ ufshcd_pci_probe(struct pci_dev *pdev, const struct pci_device_id *id)
 
         mmio_base = pcim_iomap_table(pdev)[0];
 
-        err = ufshcd_init(&pdev->dev, &hba, mmio_base, pdev->irq);
+        err = ufshcd_alloc_host(&pdev->dev, &hba);
+        if (err) {
+                dev_err(&pdev->dev, "Allocation failed\n");
+                return err;
+        }
+
+        err = ufshcd_init(hba, mmio_base, pdev->irq);
         if (err) {
                 dev_err(&pdev->dev, "Initialization failed\n");
                 return err;