/**
  * adis16400_spi_read_burst() - read all data registers
- * @dev: device associated with child of actual device (iio_dev or iio_trig)
+ * @indio_dev: the IIO device
  * @rx: somewhere to pass back the value read (min size is 24 bytes)
  **/
-static int adis16400_spi_read_burst(struct device *dev, u8 *rx)
+static int adis16400_spi_read_burst(struct iio_dev *indio_dev, u8 *rx)
 {
        struct spi_message msg;
-       struct iio_dev *indio_dev = dev_get_drvdata(dev);
        struct adis16400_state *st = iio_priv(indio_dev);
        u32 old_speed_hz = st->us->max_speed_hz;
        int ret;
        cpu_to_be16(ADIS16400_READ_REG(ADIS16400_AUX_ADC)),
 };
 
-static int adis16350_spi_read_all(struct device *dev, u8 *rx)
+static int adis16350_spi_read_all(struct iio_dev *indio_dev, u8 *rx)
 {
-       struct iio_dev *indio_dev = dev_get_drvdata(dev);
        struct adis16400_state *st = iio_priv(indio_dev);
 
        struct spi_message msg;
 
        if (scan_count) {
                if (st->variant->flags & ADIS16400_NO_BURST) {
-                       ret = adis16350_spi_read_all(&indio_dev->dev, st->rx);
+                       ret = adis16350_spi_read_all(indio_dev, st->rx);
                        if (ret < 0)
                                goto err;
                        for (; i < scan_count; i++)
                                data[i] = *(s16 *)(st->rx + i*2);
                } else {
-                       ret = adis16400_spi_read_burst(&indio_dev->dev, st->rx);
+                       ret = adis16400_spi_read_burst(indio_dev, st->rx);
                        if (ret < 0)
                                goto err;
                        for (; i < scan_count; i++) {