2 files changed, 3 insertions, 2 deletions

diff --git a/include/asm-i386/alternative.h b/include/asm-i386/alternative.h
index 96adbabec740..bb3b6317c810 100644
--- a/include/asm-i386/alternative.h
+++ b/include/asm-i386/alternative.h
@@ -116,7 +116,7 @@ static inline void alternatives_smp_switch(int smp) {}
                       "  .align 4\n"                                    \
                       "  .long 661b\n"            /* label */           \
                       "  .long 663f\n"            /* new instruction */ \
-                      "  .byte 0x68\n"            /* X86_FEATURE_UP */  \
+                      "  .byte " __stringify(X86_FEATURE_UP) "\n"       \
                       "  .byte 662b-661b\n"       /* sourcelen */       \
                       "  .byte 664f-663f\n"       /* replacementlen */  \
                       ".previous\n"                                     \
diff --git a/include/asm-x86_64/alternative.h b/include/asm-x86_64/alternative.h
index aa67bfd1b3ce..709ad3f0d354 100644
--- a/include/asm-x86_64/alternative.h
+++ b/include/asm-x86_64/alternative.h
@@ -4,6 +4,7 @@
 #ifdef __KERNEL__
 
 #include <linux/types.h>
+#include <asm/cpufeature.h>
 
 struct alt_instr {
         u8 *instr;              /* original instruction */
@@ -130,7 +131,7 @@ static inline void alternatives_smp_switch(int smp) {}
                       "  .align 8\n"                                    \
                       "  .quad 661b\n"            /* label */           \
                       "  .quad 663f\n"            /* new instruction */ \
-                      "  .byte 0x66\n"            /* X86_FEATURE_UP */  \
+                      "  .byte " __stringify(X86_FEATURE_UP) "\n"       \
                       "  .byte 662b-661b\n"       /* sourcelen */       \
                       "  .byte 664f-663f\n"       /* replacementlen */  \
                       ".previous\n"                                     \