diff --git a/board/cm4008/flash.c b/board/cm4008/flash.c
index d6fd5195d30c4fa93bc3d196272831eed9a71826..2e1356f9f4a6d6f62676d0a96888ba99c4009b14 100644
--- a/board/cm4008/flash.c
+++ b/board/cm4008/flash.c
@@ -72,7 +72,7 @@ unsigned long flash_init (void)
 	 */
 	flash_protect (FLAG_PROTECT_SET,
 		       CONFIG_SYS_FLASH_BASE,
-		       CONFIG_SYS_FLASH_BASE + _bss_start - _armboot_start,
+		       CONFIG_SYS_FLASH_BASE + _bss_start_ofs,
 		       &flash_info[0]);
 
 	return size;
diff --git a/board/cm41xx/flash.c b/board/cm41xx/flash.c
index d6fd5195d30c4fa93bc3d196272831eed9a71826..2e1356f9f4a6d6f62676d0a96888ba99c4009b14 100644
--- a/board/cm41xx/flash.c
+++ b/board/cm41xx/flash.c
@@ -72,7 +72,7 @@ unsigned long flash_init (void)
 	 */
 	flash_protect (FLAG_PROTECT_SET,
 		       CONFIG_SYS_FLASH_BASE,
-		       CONFIG_SYS_FLASH_BASE + _bss_start - _armboot_start,
+		       CONFIG_SYS_FLASH_BASE + _bss_start_ofs,
 		       &flash_info[0]);
 
 	return size;