#include <linux/module.h>
 #include <linux/delay.h>
+#include <linux/devm-helpers.h>
 #include <linux/err.h>
 #include <linux/kernel.h>
 #include <linux/interrupt.h>
                        return ret;
                }
 
-               INIT_DELAYED_WORK(&vreg->ocp_work, spmi_regulator_vs_ocp_work);
+               ret = devm_delayed_work_autocancel(dev, &vreg->ocp_work,
+                                                  spmi_regulator_vs_ocp_work);
+               if (ret)
+                       return ret;
        }
 
        return 0;
                vreg->regmap = regmap;
                if (reg->ocp) {
                        vreg->ocp_irq = platform_get_irq_byname(pdev, reg->ocp);
-                       if (vreg->ocp_irq < 0) {
-                               ret = vreg->ocp_irq;
-                               goto err;
-                       }
+                       if (vreg->ocp_irq < 0)
+                               return vreg->ocp_irq;
                }
                vreg->desc.id = -1;
                vreg->desc.owner = THIS_MODULE;
                rdev = devm_regulator_register(dev, &vreg->desc, &config);
                if (IS_ERR(rdev)) {
                        dev_err(dev, "failed to register %s\n", name);
-                       ret = PTR_ERR(rdev);
-                       goto err;
+                       return PTR_ERR(rdev);
                }
 
                INIT_LIST_HEAD(&vreg->node);
        }
 
        return 0;
-
-err:
-       list_for_each_entry(vreg, vreg_list, node)
-               if (vreg->ocp_irq)
-                       cancel_delayed_work_sync(&vreg->ocp_work);
-       return ret;
-}
-
-static int qcom_spmi_regulator_remove(struct platform_device *pdev)
-{
-       struct spmi_regulator *vreg;
-       struct list_head *vreg_list = platform_get_drvdata(pdev);
-
-       list_for_each_entry(vreg, vreg_list, node)
-               if (vreg->ocp_irq)
-                       cancel_delayed_work_sync(&vreg->ocp_work);
-
-       return 0;
 }
 
 static struct platform_driver qcom_spmi_regulator_driver = {
                .of_match_table = qcom_spmi_regulator_match,
        },
        .probe          = qcom_spmi_regulator_probe,
-       .remove         = qcom_spmi_regulator_remove,
 };
 module_platform_driver(qcom_spmi_regulator_driver);