staging:iio: fix removal path to allow correct freeing.

[deliverable/linux.git] / drivers / staging / iio / meter / ade7759.c

diff --git a/drivers/staging/iio/meter/ade7759.c b/drivers/staging/iio/meter/ade7759.c
index 511a60500d93ecb8940d6b1b7f5ea4ac097d30bd..b691f10ce988c90c7974d527b085a5defb122d6b 100644 (file)
--- a/drivers/staging/iio/meter/ade7759.c
+++ b/drivers/staging/iio/meter/ade7759.c
@@ -502,11 +502,12 @@ static int ade7759_remove(struct spi_device *spi)
        int ret;
        struct iio_dev *indio_dev = spi_get_drvdata(spi);
 
+       iio_device_unregister(indio_dev);
        ret = ade7759_stop_device(&(indio_dev->dev));
        if (ret)
                goto err_ret;
 
-       iio_device_unregister(indio_dev);
+       iio_free_device(indio_dev);
 
 err_ret:
        return ret;