if (!rk_dailink.cpu_of_node) {
                dev_err(&pdev->dev,
                        "Property 'rockchip,i2s-controller' missing or invalid\n");
-               return -EINVAL;
+               ret = -EINVAL;
+               goto put_codec_of_node;
        }
 
        rk_dailink.platform_of_node = rk_dailink.cpu_of_node;
        if (ret) {
                dev_err(&pdev->dev,
                        "Soc parse card name failed %d\n", ret);
-               return ret;
+               goto put_cpu_of_node;
        }
 
        ret = devm_snd_soc_register_card(&pdev->dev, card);
        if (ret) {
                dev_err(&pdev->dev,
                        "Soc register card failed %d\n", ret);
-               return ret;
+               goto put_cpu_of_node;
        }
 
        return ret;
+
+put_cpu_of_node:
+       of_node_put(rk_dailink.cpu_of_node);
+       rk_dailink.cpu_of_node = NULL;
+put_codec_of_node:
+       of_node_put(rk_dailink.codec_of_node);
+       rk_dailink.codec_of_node = NULL;
+
+       return ret;
+}
+
+static int snd_rk_mc_remove(struct platform_device *pdev)
+{
+       of_node_put(rk_dailink.cpu_of_node);
+       rk_dailink.cpu_of_node = NULL;
+       of_node_put(rk_dailink.codec_of_node);
+       rk_dailink.codec_of_node = NULL;
+
+       return 0;
 }
 
 static const struct of_device_id rockchip_rt5645_of_match[] = {
 
 static struct platform_driver snd_rk_mc_driver = {
        .probe = snd_rk_mc_probe,
+       .remove = snd_rk_mc_remove,
        .driver = {
                .name = DRV_NAME,
                .pm = &snd_soc_pm_ops,