if (ret)
                return ret;
 
-       ret = iio_device_register(indio_dev);
+       ret = devm_iio_device_register(dev, indio_dev);
        if (ret) {
                dev_err(&pdev->dev, "Couldn't register the device.\n");
                return ret;
        return 0;
 }
 
-static int imx7d_adc_remove(struct platform_device *pdev)
-{
-       struct iio_dev *indio_dev = platform_get_drvdata(pdev);
-
-       iio_device_unregister(indio_dev);
-
-       return 0;
-}
-
-
 static SIMPLE_DEV_PM_OPS(imx7d_adc_pm_ops, imx7d_adc_disable, imx7d_adc_enable);
 
 static struct platform_driver imx7d_adc_driver = {
        .probe          = imx7d_adc_probe,
-       .remove         = imx7d_adc_remove,
        .driver         = {
                .name   = "imx7d_adc",
                .of_match_table = imx7d_adc_match,