Regulators: pcap-regulator - clean up driver data after removal

It is a good tone to reset driver data after unbinding the device.

Signed-off-by: Dmitry Torokhov <dtor@mail.ru>
Acked-by: Mark Brown <broonie@opensource.wolfsonmicro.com>
Signed-off-by: Liam Girdwood <lrg@slimlogic.co.uk>

1 files changed, 5 insertions, 3 deletions

diff --git a/drivers/regulator/pcap-regulator.c b/drivers/regulator/pcap-regulator.c
index 33d7d899e030..29d0566379ae 100644
--- a/drivers/regulator/pcap-regulator.c
+++ b/drivers/regulator/pcap-regulator.c
@@ -288,16 +288,18 @@ static int __devexit pcap_regulator_remove(struct platform_device *pdev)
         struct regulator_dev *rdev = platform_get_drvdata(pdev);
 
         regulator_unregister(rdev);
+        platform_set_drvdata(pdev, NULL);
 
         return 0;
 }
 
 static struct platform_driver pcap_regulator_driver = {
         .driver = {
-                .name = "pcap-regulator",
+                .name   = "pcap-regulator",
+                .owner  = THIS_MODULE,
         },
-        .probe = pcap_regulator_probe,
-        .remove = __devexit_p(pcap_regulator_remove),
+        .probe  = pcap_regulator_probe,
+        .remove = __devexit_p(pcap_regulator_remove),
 };
 
 static int __init pcap_regulator_init(void)