diff --git a/scripts/wk/hw/diags.py b/scripts/wk/hw/diags.py
index 3716fe3d..070351ff 100644
--- a/scripts/wk/hw/diags.py
+++ b/scripts/wk/hw/diags.py
@@ -290,7 +290,9 @@ def build_menu(cli_mode=False, quick_mode=False) -> cli.Menu:
 
   # Skip CPU tests for TestStations
   if os.path.exists(cfg.hw.TESTSTATION_FILE):
-    menu.options['CPU & Cooling']['Selected'] = False
+    menu.options['CPU (Sysbench)']['Selected'] = False
+    menu.options['CPU (Prime95)']['Selected'] = False
+    menu.options['CPU (Cooling)']['Selected'] = False
 
   # Add CLI actions if necessary
   if cli_mode or 'DISPLAY' not in os.environ: