1 files changed, 2 insertions, 3 deletions

diff --git a/drivers/power/da9052-battery.c b/drivers/power/da9052-battery.c
index 08193feb3b08..f8f4c0f7c17d 100644
--- a/drivers/power/da9052-battery.c
+++ b/drivers/power/da9052-battery.c
@@ -594,7 +594,8 @@ static s32 da9052_bat_probe(struct platform_device *pdev)
         int ret;
         int i;
 
-        bat = kzalloc(sizeof(struct da9052_battery), GFP_KERNEL);
+        bat = devm_kzalloc(&pdev->dev, sizeof(struct da9052_battery),
+                                GFP_KERNEL);
         if (!bat)
                 return -ENOMEM;
 
@@ -635,7 +636,6 @@ err:
         while (--i >= 0)
                 da9052_free_irq(bat->da9052, da9052_bat_irq_bits[i], bat);
 
-        kfree(bat);
         return ret;
 }
 static int da9052_bat_remove(struct platform_device *pdev)
@@ -647,7 +647,6 @@ static int da9052_bat_remove(struct platform_device *pdev)
                 da9052_free_irq(bat->da9052, da9052_bat_irq_bits[i], bat);
 
         power_supply_unregister(&bat->psy);
-        kfree(bat);
 
         return 0;
 }