2 files changed, 6 insertions, 4 deletions

diff --git a/drivers/iio/buffer_cb.c b/drivers/iio/buffer_cb.c
index 4d40e24f3721..9201022945e9 100644
--- a/drivers/iio/buffer_cb.c
+++ b/drivers/iio/buffer_cb.c
@@ -25,7 +25,7 @@ static struct iio_buffer_access_funcs iio_cb_access = {
         .store_to = &iio_buffer_cb_store_to,
 };
 
-struct iio_cb_buffer *iio_channel_get_all_cb(const char *name,
+struct iio_cb_buffer *iio_channel_get_all_cb(struct device *dev,
                                              int (*cb)(u8 *data,
                                                        void *private),
                                              void *private)
@@ -46,7 +46,7 @@ struct iio_cb_buffer *iio_channel_get_all_cb(const char *name,
         cb_buff->buffer.access = &iio_cb_access;
         INIT_LIST_HEAD(&cb_buff->buffer.demux_list);
 
-        cb_buff->channels = iio_channel_get_all(name);
+        cb_buff->channels = iio_channel_get_all(dev);
         if (IS_ERR(cb_buff->channels)) {
                 ret = PTR_ERR(cb_buff->channels);
                 goto error_free_cb_buff;
diff --git a/drivers/iio/inkern.c b/drivers/iio/inkern.c
index d55e98fb300e..58d0ffe856b6 100644
--- a/drivers/iio/inkern.c
+++ b/drivers/iio/inkern.c
@@ -167,16 +167,18 @@ void iio_channel_release(struct iio_channel *channel)
 }
 EXPORT_SYMBOL_GPL(iio_channel_release);
 
-struct iio_channel *iio_channel_get_all(const char *name)
+struct iio_channel *iio_channel_get_all(struct device *dev)
 {
+        const char *name;
         struct iio_channel *chans;
         struct iio_map_internal *c = NULL;
         int nummaps = 0;
         int mapind = 0;
         int i, ret;
 
-        if (name == NULL)
+        if (dev == NULL)
                 return ERR_PTR(-EINVAL);
+        name = dev_name(dev);
 
         mutex_lock(&iio_map_list_lock);
         /* first count the matching maps */