diff --git a/pi-test-status/extensions/index.ts b/pi-test-status/extensions/index.ts
index 2fcc04e..7ed8a77 100644
--- a/pi-test-status/extensions/index.ts
+++ b/pi-test-status/extensions/index.ts
@@ -1,4 +1,4 @@
-import {type ExtensionAPI, isBashToolResult} from "@earendil-works/pi-coding-agent";
+import {type ExtensionAPI, isBashToolResult, ThemeColor} from "@earendil-works/pi-coding-agent";
 import {isTestCommand, parseTestOutput} from "./parsing/typescript-command-output";
 
 // noinspection JSUnusedGlobalSymbols
@@ -15,13 +15,14 @@ export default function testStatusExtension(pi: ExtensionAPI) {
 
         const testResult = parseTestOutput(output);
 
+        const feedbackTheme : ThemeColor = (testResult.status === "Success") ? "success" : "error";
         if (testResult.status === "Success") {
             ctx.ui.setWidget("test-status", [
-                `${theme.fg("success", "●")} ${theme.fg("text", testResult.summary)}`,
+                `${theme.fg(feedbackTheme,  "●")} ${theme.fg("text", testResult.summary)}`,
             ]);
         } else if (testResult.status === "Fail" || event.isError) {
             ctx.ui.setWidget("test-status", [
-                `${theme.fg("error", "●")} ${theme.fg("text", testResult.summary)}`,
+                `${theme.fg(feedbackTheme, "●")} ${theme.fg("text", testResult.summary)}`,
             ]);
         }
     });