.num_dacs = 4,
 };
 
-static const struct of_device_id wm8580_of_match[] = {
-       { .compatible = "wlf,wm8580", .data = &wm8580_data },
-       { .compatible = "wlf,wm8581", .data = &wm8581_data },
-       { },
-};
-MODULE_DEVICE_TABLE(of, wm8580_of_match);
-
 static int wm8580_i2c_probe(struct i2c_client *i2c)
 {
-       const struct of_device_id *of_id;
        struct wm8580_priv *wm8580;
        int ret, i;
 
 
        i2c_set_clientdata(i2c, wm8580);
 
-       of_id = of_match_device(wm8580_of_match, &i2c->dev);
-       if (of_id)
-               wm8580->drvdata = of_id->data;
-
-       if (!wm8580->drvdata) {
-               dev_err(&i2c->dev, "failed to find driver data\n");
-               return -EINVAL;
-       }
+       wm8580->drvdata = i2c_get_match_data(i2c);
+       if (!wm8580->drvdata)
+               return dev_err_probe(&i2c->dev, -EINVAL, "failed to find driver data\n");
 
        ret = devm_snd_soc_register_component(&i2c->dev,
                        &soc_component_dev_wm8580, wm8580_dai, ARRAY_SIZE(wm8580_dai));
        return ret;
 }
 
+static const struct of_device_id wm8580_of_match[] = {
+       { .compatible = "wlf,wm8580", .data = &wm8580_data },
+       { .compatible = "wlf,wm8581", .data = &wm8581_data },
+       { }
+};
+MODULE_DEVICE_TABLE(of, wm8580_of_match);
+
 static const struct i2c_device_id wm8580_i2c_id[] = {
        { "wm8580", (kernel_ulong_t)&wm8580_data },
        { "wm8581", (kernel_ulong_t)&wm8581_data },