From 0c1c65182cdd31b0a08ede4330b4536c3cf43dd2 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Wed, 12 Jul 2023 12:48:31 -0700
Subject: [PATCH] Fix HW Diags test selections for teststations

---
 scripts/wk/hw/diags.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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: