#include <linux/types.h>
 #include <linux/timer.h>
 #include <linux/watchdog.h>
-#include <linux/notifier.h>
 #include <linux/reboot.h>
 #include <linux/init.h>
 #include <linux/jiffies.h>
        return 0;
 }
 
-/*
- *     Notifier for system down
- */
-
-static int softdog_notify_sys(struct notifier_block *this, unsigned long code,
-       void *unused)
-{
-       if (code == SYS_DOWN || code == SYS_HALT)
-               /* Turn the WDT off */
-               softdog_stop(NULL);
-       return NOTIFY_DONE;
-}
-
 /*
  *     Kernel Interfaces
  */
 
-static struct notifier_block softdog_notifier = {
-       .notifier_call  = softdog_notify_sys,
-};
-
 static struct watchdog_info softdog_info = {
        .identity = "Software Watchdog",
        .options = WDIOF_SETTIMEOUT | WDIOF_KEEPALIVEPING | WDIOF_MAGICCLOSE,
        softdog_dev.timeout = soft_margin;
 
        watchdog_set_nowayout(&softdog_dev, nowayout);
-
-       ret = register_reboot_notifier(&softdog_notifier);
-       if (ret) {
-               pr_err("cannot register reboot notifier (err=%d)\n", ret);
-               return ret;
-       }
+       watchdog_stop_on_reboot(&softdog_dev);
 
        ret = watchdog_register_device(&softdog_dev);
-       if (ret) {
-               unregister_reboot_notifier(&softdog_notifier);
+       if (ret)
                return ret;
-       }
 
        pr_info("Software Watchdog Timer: 0.08 initialized. soft_noboot=%d soft_margin=%d sec soft_panic=%d (nowayout=%d)\n",
                soft_noboot, soft_margin, soft_panic, nowayout);
 static void __exit watchdog_exit(void)
 {
        watchdog_unregister_device(&softdog_dev);
-       unregister_reboot_notifier(&softdog_notifier);
 }
 
 module_init(watchdog_init);