diff --git a/src/mainboard/google/rambi/chromeos.c b/src/mainboard/google/rambi/chromeos.c
index a225821fd8..50497f07aa 100644
--- a/src/mainboard/google/rambi/chromeos.c
+++ b/src/mainboard/google/rambi/chromeos.c
@@ -31,7 +31,7 @@ int get_write_protect_state(void)
 	 * there is a 10K pullup. Disable the internal pull in romstage so that
 	 * there isn't any ambiguity in the reading.
 	 */
-	if (ENV_ROMSTAGE)
+	if (ENV_ROMSTAGE_OR_BEFORE)
 		ssus_disable_internal_pull(WP_STATUS_PAD);
 
 	/* WP is enabled when the pin is reading high. */