diff --git a/tests/test_gui_progress.py b/tests/test_gui_progress.py
index 56592e29..104351c6 100644
--- a/tests/test_gui_progress.py
+++ b/tests/test_gui_progress.py
@@ -24,6 +24,7 @@ def test_render_mma_dashboard_progress():
  # Patch where it is actually used
  with patch('src.gui_2.imgui', mock_imgui), \
       patch('src.imgui_scopes.imgui', new=mock_imgui), \
+      patch('src.theme_2.imgui', new=mock_imgui), \
       patch('src.gui_2.cost_tracker.estimate_cost', return_value=0.0):
   
   # Mock App instance - no spec because of delegation
@@ -77,11 +78,11 @@ def test_render_mma_dashboard_progress():
   mock_imgui.progress_bar.assert_any_call(0.25, (-1.0, 0.0), "25.0%")
   
   # Verify status breakdown counts are asserted via imgui.text_colored calls
-  mock_imgui.text_colored.assert_any_call(C_LBL, "Completed:")
-  mock_imgui.text_colored.assert_any_call(C_VAL, "1")
-  mock_imgui.text_colored.assert_any_call(C_LBL, "In Progress:")
-  mock_imgui.text_colored.assert_any_call(C_VAL, "1")
-  mock_imgui.text_colored.assert_any_call(C_LBL, "Blocked:")
-  mock_imgui.text_colored.assert_any_call(C_VAL, "1")
-  mock_imgui.text_colored.assert_any_call(C_LBL, "Todo:")
-  mock_imgui.text_colored.assert_any_call(C_VAL, "1")
+  mock_imgui.text_colored.assert_any_call(C_LBL(), "Completed:")
+  mock_imgui.text_colored.assert_any_call(C_VAL(), "1")
+  mock_imgui.text_colored.assert_any_call(C_LBL(), "In Progress:")
+  mock_imgui.text_colored.assert_any_call(C_VAL(), "1")
+  mock_imgui.text_colored.assert_any_call(C_LBL(), "Blocked:")
+  mock_imgui.text_colored.assert_any_call(C_VAL(), "1")
+  mock_imgui.text_colored.assert_any_call(C_LBL(), "Todo:")
+  mock_imgui.text_colored.assert_any_call(C_VAL(), "1")