return max;
 }
 
-static int atmel_sha204a_probe(struct i2c_client *client,
-                              const struct i2c_device_id *id)
+static int atmel_sha204a_probe(struct i2c_client *client)
 {
+       const struct i2c_device_id *id = i2c_client_get_device_id(client);
        struct atmel_i2c_client_priv *i2c_priv;
        int ret;
 
 MODULE_DEVICE_TABLE(i2c, atmel_sha204a_id);
 
 static struct i2c_driver atmel_sha204a_driver = {
-       .probe                  = atmel_sha204a_probe,
+       .probe_new              = atmel_sha204a_probe,
        .remove                 = atmel_sha204a_remove,
        .id_table               = atmel_sha204a_id,