unsigned int ctrl = 0;
        int ret;
 
-       indio_dev = iio_device_alloc(sizeof(*st));
+       indio_dev = devm_iio_device_alloc(dev, sizeof(*st));
        if (indio_dev == NULL) {
                dev_err(dev, "Failed to allocate iio device\n");
-               ret = -ENOMEM;
-               goto error_out;
+               return -ENOMEM;
        }
 
        st = iio_priv(indio_dev);
        ret = ad5380_alloc_channels(indio_dev);
        if (ret) {
                dev_err(dev, "Failed to allocate channel spec: %d\n", ret);
-               goto error_free;
+               return ret;
        }
 
        if (st->chip_info->int_vref == 2500000)
                ctrl |= AD5380_CTRL_INT_VREF_2V5;
 
-       st->vref_reg = regulator_get(dev, "vref");
+       st->vref_reg = devm_regulator_get(dev, "vref");
        if (!IS_ERR(st->vref_reg)) {
                ret = regulator_enable(st->vref_reg);
                if (ret) {
        if (!IS_ERR(st->vref_reg))
                regulator_disable(st->vref_reg);
 error_free_reg:
-       if (!IS_ERR(st->vref_reg))
-               regulator_put(st->vref_reg);
-
        kfree(indio_dev->channels);
-error_free:
-       iio_device_free(indio_dev);
-error_out:
 
        return ret;
 }
 
        if (!IS_ERR(st->vref_reg)) {
                regulator_disable(st->vref_reg);
-               regulator_put(st->vref_reg);
        }
 
-       iio_device_free(indio_dev);
-
        return 0;
 }