}
 #endif
 
-static int max8952_pmic_probe(struct i2c_client *client,
-               const struct i2c_device_id *i2c_id)
+static int max8952_pmic_probe(struct i2c_client *client)
 {
        struct i2c_adapter *adapter = client->adapter;
        struct max8952_platform_data *pdata = dev_get_platdata(&client->dev);
 MODULE_DEVICE_TABLE(i2c, max8952_ids);
 
 static struct i2c_driver max8952_pmic_driver = {
-       .probe          = max8952_pmic_probe,
+       .probe_new      = max8952_pmic_probe,
        .driver         = {
                .name   = "max8952",
                .of_match_table = of_match_ptr(max8952_dt_match),