diff --git a/src/console/console.c b/src/console/console.c
index 9a079ed..bb9d374 100644
--- a/src/console/console.c
+++ b/src/console/console.c
@@ -86,7 +86,7 @@
 		console_tx_byte(*buffer++);
 }
 
-#if CONFIG(GDB_STUB) && (ENV_ROMSTAGE || ENV_RAMSTAGE)
+#if CONFIG(GDB_STUB) && (ENV_ROMSTAGE_OR_BEFORE || ENV_RAMSTAGE)
 void gdb_hw_init(void)
 {
 	__gdb_hw_init();
diff --git a/src/include/console/uart.h b/src/include/console/uart.h
index 2e23d43..ca03ecb 100644
--- a/src/include/console/uart.h
+++ b/src/include/console/uart.h
@@ -84,7 +84,7 @@
 static inline void __uart_tx_flush(void)	{}
 #endif
 
-#if CONFIG(GDB_STUB) && (ENV_ROMSTAGE || ENV_RAMSTAGE)
+#if CONFIG(GDB_STUB) && (ENV_ROMSTAGE_OR_BEFORE || ENV_RAMSTAGE)
 #define CONF_UART_FOR_GDB	CONFIG_UART_FOR_CONSOLE
 static inline void __gdb_hw_init(void)	{ uart_init(CONF_UART_FOR_GDB); }
 static inline void __gdb_tx_byte(u8 data)