indio_dev->num_channels = ARRAY_SIZE(ad2s1210_channels);
        indio_dev->name = spi_get_device_id(spi)->name;
 
-       ret = iio_device_register(indio_dev);
+       ret = devm_iio_device_register(&spi->dev, indio_dev);
        if (ret)
                return ret;
 
        return 0;
 }
 
-static int ad2s1210_remove(struct spi_device *spi)
-{
-       struct iio_dev *indio_dev = spi_get_drvdata(spi);
-
-       iio_device_unregister(indio_dev);
-
-       return 0;
-}
-
 static const struct of_device_id ad2s1210_of_match[] = {
        { .compatible = "adi,ad2s1210", },
        { }
                .of_match_table = of_match_ptr(ad2s1210_of_match),
        },
        .probe = ad2s1210_probe,
-       .remove = ad2s1210_remove,
        .id_table = ad2s1210_id,
 };
 module_spi_driver(ad2s1210_driver);