1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/spi/spi-axi-spi-engine.c b/drivers/spi/spi-axi-spi-engine.c
index 6ab4c7700228..68cfc351b47f 100644
--- a/drivers/spi/spi-axi-spi-engine.c
+++ b/drivers/spi/spi-axi-spi-engine.c
@@ -553,7 +553,7 @@ err_put_master:
 
 static int spi_engine_remove(struct platform_device *pdev)
 {
-        struct spi_master *master = platform_get_drvdata(pdev);
+        struct spi_master *master = spi_master_get(platform_get_drvdata(pdev));
         struct spi_engine *spi_engine = spi_master_get_devdata(master);
         int irq = platform_get_irq(pdev, 0);
 
@@ -561,6 +561,8 @@ static int spi_engine_remove(struct platform_device *pdev)
 
         free_irq(irq, master);
 
+        spi_master_put(master);
+
         writel_relaxed(0xff, spi_engine->base + SPI_ENGINE_REG_INT_PENDING);
         writel_relaxed(0x00, spi_engine->base + SPI_ENGINE_REG_INT_ENABLE);
         writel_relaxed(0x01, spi_engine->base + SPI_ENGINE_REG_RESET);