2 files changed, 4 insertions, 2 deletions

diff --git a/arch/x86/boot/compressed/head_32.S b/arch/x86/boot/compressed/head_32.S
index 1d7fbbcc196d..8ef964ddc18e 100644
--- a/arch/x86/boot/compressed/head_32.S
+++ b/arch/x86/boot/compressed/head_32.S
@@ -29,6 +29,7 @@
 #include <asm/page_types.h>
 #include <asm/boot.h>
 #include <asm/asm-offsets.h>
+#include <asm/bootparam.h>
 
         __HEAD
 ENTRY(startup_32)
@@ -102,7 +103,7 @@ preferred_addr:
          * Test KEEP_SEGMENTS flag to see if the bootloader is asking
          * us to not reload segments
          */
-        testb   $(1<<6), BP_loadflags(%esi)
+        testb   $KEEP_SEGMENTS, BP_loadflags(%esi)
         jnz     1f
 
         cli
diff --git a/arch/x86/boot/compressed/head_64.S b/arch/x86/boot/compressed/head_64.S
index 6b1766c6c082..90c152135e92 100644
--- a/arch/x86/boot/compressed/head_64.S
+++ b/arch/x86/boot/compressed/head_64.S
@@ -31,6 +31,7 @@
 #include <asm/msr.h>
 #include <asm/processor-flags.h>
 #include <asm/asm-offsets.h>
+#include <asm/bootparam.h>
 
         __HEAD
         .code32
@@ -46,7 +47,7 @@ ENTRY(startup_32)
          * Test KEEP_SEGMENTS flag to see if the bootloader is asking
          * us to not reload segments
          */
-        testb $(1<<6), BP_loadflags(%esi)
+        testb $KEEP_SEGMENTS, BP_loadflags(%esi)
         jnz 1f
 
         cli