AK4613_CONFIG_SDTI_set(priv, sdti_num);
 }
 
-static int ak4613_i2c_probe(struct i2c_client *i2c,
-                           const struct i2c_device_id *id)
+static int ak4613_i2c_probe(struct i2c_client *i2c)
 {
        struct device *dev = &i2c->dev;
        struct device_node *np = dev->of_node;
        regmap_cfg = NULL;
        if (np)
                regmap_cfg = of_device_get_match_data(dev);
-       else
+       else {
+               const struct i2c_device_id *id =
+                       i2c_match_id(ak4613_i2c_id, i2c);
                regmap_cfg = (const struct regmap_config *)id->driver_data;
+       }
 
        if (!regmap_cfg)
                return -EINVAL;
                .name = "ak4613-codec",
                .of_match_table = ak4613_of_match,
        },
-       .probe          = ak4613_i2c_probe,
+       .probe_new      = ak4613_i2c_probe,
        .remove         = ak4613_i2c_remove,
        .id_table       = ak4613_i2c_id,
 };
 
 #endif
 
 static const struct of_device_id ak4642_of_match[];
-static int ak4642_i2c_probe(struct i2c_client *i2c,
-                           const struct i2c_device_id *id)
+static const struct i2c_device_id ak4642_i2c_id[];
+static int ak4642_i2c_probe(struct i2c_client *i2c)
 {
        struct device *dev = &i2c->dev;
        struct device_node *np = dev->of_node;
                if (of_id)
                        drvdata = of_id->data;
        } else {
+               const struct i2c_device_id *id =
+                       i2c_match_id(ak4642_i2c_id, i2c);
                drvdata = (const struct ak4642_drvdata *)id->driver_data;
        }
 
                .name = "ak4642-codec",
                .of_match_table = ak4642_of_match,
        },
-       .probe          = ak4642_i2c_probe,
+       .probe_new      = ak4642_i2c_probe,
        .id_table       = ak4642_i2c_id,
 };