1 files changed, 9 insertions, 12 deletions

diff --git a/drivers/watchdog/acquirewdt.c b/drivers/watchdog/acquirewdt.c
index 5cf1621def9c..5614416f1032 100644
--- a/drivers/watchdog/acquirewdt.c
+++ b/drivers/watchdog/acquirewdt.c
@@ -239,7 +239,7 @@ static struct miscdevice acq_miscdev = {
  *      Init & exit routines
  */
 
-static int acq_probe(struct platform_device *dev)
+static int __init acq_probe(struct platform_device *dev)
 {
         int ret;
 
@@ -291,7 +291,6 @@ static void acq_shutdown(struct platform_device *dev)
 }
 
 static struct platform_driver acquirewdt_driver = {
-        .probe          = acq_probe,
         .remove         = acq_remove,
         .shutdown       = acq_shutdown,
         .driver         = {
@@ -306,20 +305,18 @@ static int __init acq_init(void)
 
         pr_info("WDT driver for Acquire single board computer initialising\n");
 
-        err = platform_driver_register(&acquirewdt_driver);
-        if (err)
-                return err;
-
         acq_platform_device = platform_device_register_simple(DRV_NAME,
                                                                 -1, NULL, 0);
-        if (IS_ERR(acq_platform_device)) {
-                err = PTR_ERR(acq_platform_device);
-                goto unreg_platform_driver;
-        }
+        if (IS_ERR(acq_platform_device))
+                return PTR_ERR(acq_platform_device);
+
+        err = platform_driver_probe(&acquirewdt_driver, acq_probe);
+        if (err)
+                goto unreg_platform_device;
         return 0;
 
-unreg_platform_driver:
-        platform_driver_unregister(&acquirewdt_driver);
+unreg_platform_device:
+        platform_device_unregister(acq_platform_device);
         return err;
 }