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

[deliverable/linux.git] / drivers / staging / iio / dac / ad5504.c

diff --git a/drivers/staging/iio/dac/ad5504.c b/drivers/staging/iio/dac/ad5504.c
index c33643f28cb00aeb647ca78fa5ce279e677a3509..60dd6404d689d2dafa0805eb132432588475fc79 100644 (file)
--- a/drivers/staging/iio/dac/ad5504.c
+++ b/drivers/staging/iio/dac/ad5504.c
@@ -349,6 +349,7 @@ static int __devexit ad5504_remove(struct spi_device *spi)
        struct iio_dev *indio_dev = spi_get_drvdata(spi);
        struct ad5504_state *st = iio_priv(indio_dev);
 
+       iio_device_unregister(indio_dev);
        if (spi->irq)
                free_irq(spi->irq, indio_dev);
 
@@ -356,8 +357,7 @@ static int __devexit ad5504_remove(struct spi_device *spi)
                regulator_disable(st->reg);
                regulator_put(st->reg);
        }
-
-       iio_device_unregister(indio_dev);
+       iio_free_device(indio_dev);
 
        return 0;
 }