1 files changed, 4 insertions, 14 deletions

diff --git a/drivers/watchdog/via_wdt.c b/drivers/watchdog/via_wdt.c
index 465e08273c97..5603e31afdab 100644
--- a/drivers/watchdog/via_wdt.c
+++ b/drivers/watchdog/via_wdt.c
@@ -202,6 +202,9 @@ static int __devinit wdt_probe(struct pci_dev *pdev,
                 goto err_out_release;
         }
 
+        if (timeout < 1 || timeout > WDT_TIMEOUT_MAX)
+                timeout = WDT_TIMEOUT;
+
         wdt_dev.timeout = timeout;
         watchdog_set_nowayout(&wdt_dev, nowayout);
         if (readl(wdt_mem) & VIA_WDT_FIRED)
@@ -250,20 +253,7 @@ static struct pci_driver wdt_driver = {
         .remove         = __devexit_p(wdt_remove),
 };
 
-static int __init wdt_init(void)
-{
-        if (timeout < 1 || timeout > WDT_TIMEOUT_MAX)
-                timeout = WDT_TIMEOUT;
-        return pci_register_driver(&wdt_driver);
-}
-
-static void __exit wdt_exit(void)
-{
-        pci_unregister_driver(&wdt_driver);
-}
-
-module_init(wdt_init);
-module_exit(wdt_exit);
+module_pci_driver(wdt_driver);
 
 MODULE_AUTHOR("Marc Vertes");
 MODULE_DESCRIPTION("Driver for watchdog timer on VIA chipset");