#include "cs42xx8.h"
 
-static const struct of_device_id cs42xx8_of_match[];
-
 static int cs42xx8_i2c_probe(struct i2c_client *i2c)
 {
        int ret;
        struct cs42xx8_driver_data *drvdata;
-       const struct of_device_id *of_id;
-
-       of_id = of_match_device(cs42xx8_of_match, &i2c->dev);
-       if (!of_id) {
-               dev_err(&i2c->dev, "failed to find driver data\n");
-               return -EINVAL;
-       }
 
-       drvdata = (struct cs42xx8_driver_data *)of_id->data;
+       drvdata = (struct cs42xx8_driver_data *)i2c_get_match_data(i2c);
+       if (!drvdata)
+               return dev_err_probe(&i2c->dev, -EINVAL,
+                                    "failed to find driver data\n");
 
        ret = cs42xx8_probe(&i2c->dev,
                devm_regmap_init_i2c(i2c, &cs42xx8_regmap_config), drvdata);