.get_property   = lp8788_battery_get_property,
 };
 
-static void lp8788_psy_unregister(struct lp8788_charger *pchg)
-{
-       power_supply_unregister(pchg->battery);
-       power_supply_unregister(pchg->charger);
-}
-
 static void lp8788_charger_event(struct work_struct *work)
 {
        struct lp8788_charger *pchg =
        charger_cfg.supplied_to = battery_supplied_to;
        charger_cfg.num_supplicants = ARRAY_SIZE(battery_supplied_to);
 
-       pchg->charger = power_supply_register(&pdev->dev,
-                                             &lp8788_psy_charger_desc,
-                                             &charger_cfg);
+       pchg->charger = devm_power_supply_register(&pdev->dev,
+                                                  &lp8788_psy_charger_desc,
+                                                  &charger_cfg);
        if (IS_ERR(pchg->charger))
                return -EPERM;
 
-       pchg->battery = power_supply_register(&pdev->dev,
-                                             &lp8788_psy_battery_desc, NULL);
-       if (IS_ERR(pchg->battery)) {
-               power_supply_unregister(pchg->charger);
+       pchg->battery = devm_power_supply_register(&pdev->dev,
+                                                  &lp8788_psy_battery_desc, NULL);
+       if (IS_ERR(pchg->battery))
                return -EPERM;
-       }
 
        return 0;
 }
 
        flush_work(&pchg->charger_work);
        lp8788_irq_unregister(pdev, pchg);
-       lp8788_psy_unregister(pchg);
 }
 
 static struct platform_driver lp8788_charger_driver = {