1 files changed, 2 insertions, 1 deletions

diff --git a/arch/x86/kernel/nmi.c b/arch/x86/kernel/nmi.c
index 19f1b95265cf..326a8f4f50f8 100644
--- a/arch/x86/kernel/nmi.c
+++ b/arch/x86/kernel/nmi.c
@@ -354,7 +354,8 @@ static DEFINE_PER_CPU(int, nmi_touch);
 
 void touch_nmi_watchdog(void)
 {
-        if (nmi_watchdog > 0) {
+        if (nmi_watchdog == NMI_LOCAL_APIC ||
+                nmi_watchdog == NMI_IO_APIC) {
                 unsigned cpu;
 
                 /*