.cache_type = REGCACHE_RBTREE,
 };
 
-static int pfuze100_regulator_probe(struct i2c_client *client,
-                                   const struct i2c_device_id *id)
+static int pfuze100_regulator_probe(struct i2c_client *client)
 {
+       const struct i2c_device_id *id = i2c_client_get_device_id(client);
        struct pfuze_chip *pfuze_chip;
        struct regulator_config config = { };
        int i, ret;
                .name = "pfuze100-regulator",
                .of_match_table = pfuze_dt_ids,
        },
-       .probe = pfuze100_regulator_probe,
+       .probe_new = pfuze100_regulator_probe,
 };
 module_i2c_driver(pfuze_driver);