diff --git a/src/arch/mips/include/arch/header.ld b/src/arch/mips/include/arch/header.ld
index 665013294c..7f832eb51c 100644
--- a/src/arch/mips/include/arch/header.ld
+++ b/src/arch/mips/include/arch/header.ld
@@ -19,7 +19,7 @@ PHDRS
 	to_load PT_LOAD;
 }
 
-#ifdef __BOOTBLOCK__
+#if ENV_BOOTBLOCK
 ENTRY(_start)
 #else
 ENTRY(stage_entry)
diff --git a/src/arch/ppc64/include/arch/header.ld b/src/arch/ppc64/include/arch/header.ld
index 349fd5165c..badeefdf49 100644
--- a/src/arch/ppc64/include/arch/header.ld
+++ b/src/arch/ppc64/include/arch/header.ld
@@ -19,7 +19,7 @@ PHDRS
 	to_load PT_LOAD;
 }
 
-#ifdef __BOOTBLOCK__
+#if ENV_BOOTBLOCK
 ENTRY(_start)
 #else
 ENTRY(stage_entry)