1 files changed, 3 insertions, 14 deletions

diff --git a/drivers/leds/leds-wm831x-status.c b/drivers/leds/leds-wm831x-status.c
index b1eb34c3e81f..74a24cf897c3 100644
--- a/drivers/leds/leds-wm831x-status.c
+++ b/drivers/leds/leds-wm831x-status.c
@@ -237,7 +237,8 @@ static int wm831x_status_probe(struct platform_device *pdev)
                 goto err;
         }
 
-        drvdata = kzalloc(sizeof(struct wm831x_status), GFP_KERNEL);
+        drvdata = devm_kzalloc(&pdev->dev, sizeof(struct wm831x_status),
+                               GFP_KERNEL);
         if (!drvdata)
                 return -ENOMEM;
         dev_set_drvdata(&pdev->dev, drvdata);
@@ -300,7 +301,6 @@ static int wm831x_status_probe(struct platform_device *pdev)
 
 err_led:
         led_classdev_unregister(&drvdata->cdev);
-        kfree(drvdata);
 err:
         return ret;
 }
@@ -311,7 +311,6 @@ static int wm831x_status_remove(struct platform_device *pdev)
 
         device_remove_file(drvdata->cdev.dev, &dev_attr_src);
         led_classdev_unregister(&drvdata->cdev);
-        kfree(drvdata);
 
         return 0;
 }
@@ -325,17 +324,7 @@ static struct platform_driver wm831x_status_driver = {
         .remove = wm831x_status_remove,
 };
 
-static int __devinit wm831x_status_init(void)
-{
-        return platform_driver_register(&wm831x_status_driver);
-}
-module_init(wm831x_status_init);
-
-static void wm831x_status_exit(void)
-{
-        platform_driver_unregister(&wm831x_status_driver);
-}
-module_exit(wm831x_status_exit);
+module_platform_driver(wm831x_status_driver);
 
 MODULE_AUTHOR("Mark Brown <broonie@opensource.wolfsonmicro.com>");
 MODULE_DESCRIPTION("WM831x status LED driver");