struct max517_platform_data *platform_data = client->dev.platform_data;
        int err;
 
-       indio_dev = iio_device_alloc(sizeof(*data));
-       if (indio_dev == NULL) {
-               err = -ENOMEM;
-               goto exit;
-       }
+       indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
+       if (!indio_dev)
+               return -ENOMEM;
        data = iio_priv(indio_dev);
        i2c_set_clientdata(client, indio_dev);
        data->client = client;
 
        err = iio_device_register(indio_dev);
        if (err)
-               goto exit_free_device;
+               return err;
 
        dev_info(&client->dev, "DAC registered\n");
 
        return 0;
-
-exit_free_device:
-       iio_device_free(indio_dev);
-exit:
-       return err;
 }
 
 static int max517_remove(struct i2c_client *client)
 {
        iio_device_unregister(i2c_get_clientdata(client));
-       iio_device_free(i2c_get_clientdata(client));
-
        return 0;
 }