*     Init & exit routines
  */
 
-static int advwdt_probe(struct platform_device *dev)
+static int __init advwdt_probe(struct platform_device *dev)
 {
        int ret;
 
 }
 
 static struct platform_driver advwdt_driver = {
-       .probe          = advwdt_probe,
        .remove         = advwdt_remove,
        .shutdown       = advwdt_shutdown,
        .driver         = {
 
        pr_info("WDT driver for Advantech single board computer initialising\n");
 
-       err = platform_driver_register(&advwdt_driver);
-       if (err)
-               return err;
-
        advwdt_platform_device = platform_device_register_simple(DRV_NAME,
                                                                -1, NULL, 0);
-       if (IS_ERR(advwdt_platform_device)) {
-               err = PTR_ERR(advwdt_platform_device);
-               goto unreg_platform_driver;
-       }
+       if (IS_ERR(advwdt_platform_device))
+               return PTR_ERR(advwdt_platform_device);
+
+       err = platform_driver_probe(&advwdt_driver, advwdt_probe);
+       if (err)
+               goto unreg_platform_device;
 
        return 0;
 
-unreg_platform_driver:
-       platform_driver_unregister(&advwdt_driver);
+unreg_platform_device:
+       platform_device_unregister(advwdt_platform_device);
        return err;
 }