static const struct st_sensor_settings st_accel_sensors_settings[] = {
        {
                .wai = ST_ACCEL_1_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LIS3DH_ACCEL_DEV_NAME,
                        [1] = LSM303DLHC_ACCEL_DEV_NAME,
        },
        {
                .wai = ST_ACCEL_2_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LIS331DLH_ACCEL_DEV_NAME,
                        [1] = LSM303DL_ACCEL_DEV_NAME,
        },
        {
                .wai = ST_ACCEL_3_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LSM330_ACCEL_DEV_NAME,
                },
        },
        {
                .wai = ST_ACCEL_4_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LIS3LV02DL_ACCEL_DEV_NAME,
                },
        },
        {
                .wai = ST_ACCEL_5_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LIS331DL_ACCEL_DEV_NAME,
                },
 
                        int num_sensors_list,
                        const struct st_sensor_settings *sensor_settings)
 {
-       u8 wai;
        int i, n, err;
+       u8 wai;
        struct st_sensor_data *sdata = iio_priv(indio_dev);
 
-       err = sdata->tf->read_byte(&sdata->tb, sdata->dev,
-                                       ST_SENSORS_DEFAULT_WAI_ADDRESS, &wai);
-       if (err < 0) {
-               dev_err(&indio_dev->dev, "failed to read Who-Am-I register.\n");
-               goto read_wai_error;
-       }
-
        for (i = 0; i < num_sensors_list; i++) {
-               if (sensor_settings[i].wai == wai)
+               for (n = 0; n < ST_SENSORS_MAX_4WAI; n++) {
+                       if (strcmp(indio_dev->name,
+                               sensor_settings[i].sensors_supported[n]) == 0) {
+                               break;
+                       }
+               }
+               if (n < ST_SENSORS_MAX_4WAI)
                        break;
        }
-       if (i == num_sensors_list)
-               goto device_not_supported;
+       if (i == num_sensors_list) {
+               dev_err(&indio_dev->dev, "device name %s not recognized.\n",
+                                                       indio_dev->name);
+               return -ENODEV;
+       }
 
-       for (n = 0; n < ARRAY_SIZE(sensor_settings[i].sensors_supported); n++) {
-               if (strcmp(indio_dev->name,
-                               &sensor_settings[i].sensors_supported[n][0]) == 0)
-                       break;
+       err = sdata->tf->read_byte(&sdata->tb, sdata->dev,
+                                       sensor_settings[i].wai_addr, &wai);
+       if (err < 0) {
+               dev_err(&indio_dev->dev, "failed to read Who-Am-I register.\n");
+               return err;
        }
-       if (n == ARRAY_SIZE(sensor_settings[i].sensors_supported)) {
-               dev_err(&indio_dev->dev, "device name \"%s\" and WhoAmI (0x%02x) mismatch",
-                       indio_dev->name, wai);
-               goto sensor_name_mismatch;
+
+       if (sensor_settings[i].wai != wai) {
+               dev_err(&indio_dev->dev, "%s: WhoAmI mismatch (0x%x).\n",
+                                               indio_dev->name, wai);
+               return -EINVAL;
        }
 
        sdata->sensor_settings =
                        (struct st_sensor_settings *)&sensor_settings[i];
 
        return i;
-
-device_not_supported:
-       dev_err(&indio_dev->dev, "device not supported: WhoAmI (0x%x).\n", wai);
-sensor_name_mismatch:
-       err = -ENODEV;
-read_wai_error:
-       return err;
 }
 EXPORT_SYMBOL(st_sensors_check_device_support);
 
 
 static const struct st_sensor_settings st_gyro_sensors_settings[] = {
        {
                .wai = ST_GYRO_1_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = L3G4200D_GYRO_DEV_NAME,
                        [1] = LSM330DL_GYRO_DEV_NAME,
        },
        {
                .wai = ST_GYRO_2_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = L3GD20_GYRO_DEV_NAME,
                        [1] = LSM330D_GYRO_DEV_NAME,
        },
        {
                .wai = ST_GYRO_3_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = L3GD20_GYRO_DEV_NAME,
                },
 
 static const struct st_sensor_settings st_magn_sensors_settings[] = {
        {
                .wai = 0, /* This sensor has no valid WhoAmI report 0 */
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LSM303DLH_MAGN_DEV_NAME,
                },
        },
        {
                .wai = ST_MAGN_1_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LSM303DLHC_MAGN_DEV_NAME,
                        [1] = LSM303DLM_MAGN_DEV_NAME,
        },
        {
                .wai = ST_MAGN_2_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LIS3MDL_MAGN_DEV_NAME,
                },
 
 static const struct st_sensor_settings st_press_sensors_settings[] = {
        {
                .wai = ST_PRESS_LPS331AP_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LPS331AP_PRESS_DEV_NAME,
                },
        },
        {
                .wai = ST_PRESS_LPS001WP_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LPS001WP_PRESS_DEV_NAME,
                },
        },
        {
                .wai = ST_PRESS_LPS25H_WAI_EXP,
+               .wai_addr = ST_SENSORS_DEFAULT_WAI_ADDRESS,
                .sensors_supported = {
                        [0] = LPS25H_PRESS_DEV_NAME,
                },
 
 /**
  * struct st_sensor_settings - ST specific sensor settings
  * @wai: Contents of WhoAmI register.
+ * @wai_addr: The address of WhoAmI register.
  * @sensors_supported: List of supported sensors by struct itself.
  * @ch: IIO channels for the sensor.
  * @odr: Output data rate register and ODR list available.
  */
 struct st_sensor_settings {
        u8 wai;
+       u8 wai_addr;
        char sensors_supported[ST_SENSORS_MAX_4WAI][ST_SENSORS_MAX_NAME];
        struct iio_chan_spec *ch;
        int num_ch;