.attrs = &nau7802_attribute_group,
 };
 
-static int nau7802_probe(struct i2c_client *client,
-                       const struct i2c_device_id *id)
+static int nau7802_probe(struct i2c_client *client)
 {
        struct iio_dev *indio_dev;
        struct nau7802_state *st;
        u8 data;
        u32 tmp = 0;
 
-       if (!client->dev.of_node) {
-               dev_err(&client->dev, "No device tree node available.\n");
-               return -EINVAL;
-       }
-
        indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*st));
        if (indio_dev == NULL)
                return -ENOMEM;
 MODULE_DEVICE_TABLE(of, nau7802_dt_ids);
 
 static struct i2c_driver nau7802_driver = {
-       .probe = nau7802_probe,
+       .probe_new = nau7802_probe,
        .id_table = nau7802_i2c_id,
        .driver = {
                   .name = "nau7802",