2 files changed, 3 insertions, 3 deletions

diff --git a/drivers/power/intel_mid_battery.c b/drivers/power/intel_mid_battery.c
index 18d136b443ee..4520811168ad 100644
--- a/drivers/power/intel_mid_battery.c
+++ b/drivers/power/intel_mid_battery.c
@@ -756,7 +756,7 @@ static int platform_pmic_battery_probe(struct platform_device *pdev)
 
 static int platform_pmic_battery_remove(struct platform_device *pdev)
 {
-        struct pmic_power_module_info *pbi = dev_get_drvdata(&pdev->dev);
+        struct pmic_power_module_info *pbi = platform_get_drvdata(pdev);
 
         free_irq(pbi->irq, pbi);
         cancel_delayed_work_sync(&pbi->monitor_battery);
diff --git a/drivers/power/tps65090-charger.c b/drivers/power/tps65090-charger.c
index 9fbca310a2ad..77ab8561fa34 100644
--- a/drivers/power/tps65090-charger.c
+++ b/drivers/power/tps65090-charger.c
@@ -218,7 +218,7 @@ static int tps65090_charger_probe(struct platform_device *pdev)
                 return -ENOMEM;
         }
 
-        dev_set_drvdata(&pdev->dev, cdata);
+        platform_set_drvdata(pdev, cdata);
 
         cdata->dev                      = &pdev->dev;
         cdata->pdata                    = pdata;
@@ -291,7 +291,7 @@ fail_unregister_supply:
 
 static int tps65090_charger_remove(struct platform_device *pdev)
 {
-        struct tps65090_charger *cdata = dev_get_drvdata(&pdev->dev);
+        struct tps65090_charger *cdata = platform_get_drvdata(pdev);
 
         devm_free_irq(cdata->dev, cdata->irq, cdata);
         power_supply_unregister(&cdata->ac);