dmaengine_terminate_all(spfi->rx_ch);
        }
        spin_unlock_irqrestore(&spfi->lock, flags);
-
-       spfi_reset(spfi);
 }
 
 static int img_spfi_prepare(struct spi_master *master, struct spi_message *msg)
        return 0;
 }
 
+static int img_spfi_unprepare(struct spi_master *master,
+                             struct spi_message *msg)
+{
+       struct img_spfi *spfi = spi_master_get_devdata(master);
+
+       spfi_reset(spfi);
+
+       return 0;
+}
+
 static void img_spfi_config(struct spi_master *master, struct spi_device *spi,
                            struct spi_transfer *xfer)
 {
        master->set_cs = img_spfi_set_cs;
        master->transfer_one = img_spfi_transfer_one;
        master->prepare_message = img_spfi_prepare;
+       master->unprepare_message = img_spfi_unprepare;
        master->handle_err = img_spfi_handle_err;
 
        spfi->tx_ch = dma_request_slave_channel(spfi->dev, "tx");