2 files changed, 2 insertions, 2 deletions

diff --git a/arch/x86/kernel/head_32.S b/arch/x86/kernel/head_32.S
index d2a21590794a..f36bd42d6f0c 100644
--- a/arch/x86/kernel/head_32.S
+++ b/arch/x86/kernel/head_32.S
@@ -545,7 +545,7 @@ ENDPROC(early_idt_handlers)
 ENTRY(early_idt_handler)
         cld
 
-        cmpl $X86_TRAP_NMI,(%esp)
+        cmpl $2,(%esp)          # X86_TRAP_NMI
         je is_nmi               # Ignore NMI
 
         cmpl $2,%ss:early_recursion_flag
diff --git a/arch/x86/kernel/head_64.S b/arch/x86/kernel/head_64.S
index 33f36c78594e..a468c0a65c42 100644
--- a/arch/x86/kernel/head_64.S
+++ b/arch/x86/kernel/head_64.S
@@ -343,7 +343,7 @@ early_idt_handlers:
 ENTRY(early_idt_handler)
         cld
 
-        cmpl $X86_TRAP_NMI,(%rsp)
+        cmpl $2,(%rsp)          # X86_TRAP_NMI
         je is_nmi               # Ignore NMI
 
         cmpl $2,early_recursion_flag(%rip)