diff --git a/scripts/wk/hw/diags.py b/scripts/wk/hw/diags.py
index f1890bf2..8958856a 100644
--- a/scripts/wk/hw/diags.py
+++ b/scripts/wk/hw/diags.py
@@ -99,7 +99,7 @@ PLATFORM = std.PLATFORM
 class State():
   # pylint: disable=too-many-instance-attributes
   """Object for tracking hardware diagnostic data."""
-  def __init__(self):
+  def __init__(self, test_mode=False):
     self.cpu_max_temp = -1
     self.disks = []
     self.layout = cfg.hw.TMUX_LAYOUT.copy()
@@ -109,6 +109,8 @@ class State():
     self.system = None
     self.test_groups = []
     self.top_text = std.color_string('Hardware Diagnostics', 'GREEN')
+    if test_mode:
+      self.top_text += std.color_string(' (Test Mode)', 'YELLOW')
 
     # Init tmux and start a background process to maintain layout
     self.init_tmux()
@@ -839,7 +841,7 @@ def main() -> None:
   # Init
   atexit.register(tmux.kill_all_panes)
   menu = build_menu(cli_mode=args['--cli'], quick_mode=args['--quick'])
-  state = State()
+  state = State(test_mode=args['--test-mode'])
   state.override_all_smart_errors = args['--ignore-smart-errors'] # pylint: disable=attribute-defined-outside-init
 
   # Quick Mode