diff --git a/src/__tests__/unit/components/BenchmarkRunsHistory.test.tsx b/src/__tests__/unit/components/BenchmarkRunsHistory.test.tsx
index 0a8928ed8c..149d34bbfa 100644
--- a/src/__tests__/unit/components/BenchmarkRunsHistory.test.tsx
+++ b/src/__tests__/unit/components/BenchmarkRunsHistory.test.tsx
@@ -867,7 +867,11 @@ describe("BenchmarkRunsHistory", () => {
       setExpandedId: mockSetExpandedId,
     });
     render(React.createElement(BenchmarkRunsHistory));
-    expect(screen.getByText("Failed")).toBeInTheDocument();
+    const failedMatches = screen.getAllByText("Failed");
+    expect(failedMatches.length).toBeGreaterThan(0);
+    expect(
+      failedMatches.some((el) => el.className.includes("text-destructive")),
+    ).toBe(true);
   });
 
   it("shows dash (not Pending) for a FAILED run with generateJamieChat (report will never fire)", () => {
diff --git a/src/__tests__/unit/components/legal/BenchmarkRunsHistory-breakdown.test.tsx b/src/__tests__/unit/components/legal/BenchmarkRunsHistory-breakdown.test.tsx
index 63b2210b0e..63612f7460 100644
--- a/src/__tests__/unit/components/legal/BenchmarkRunsHistory-breakdown.test.tsx
+++ b/src/__tests__/unit/components/legal/BenchmarkRunsHistory-breakdown.test.tsx
@@ -76,8 +76,8 @@ vi.mock("@/components/legal/HillClimbChart", () => ({
 }));
 
 vi.mock("@/components/ui/badge", () => ({
-  Badge: ({ children, className }: { children?: React.ReactNode; className?: string }) => (
-    <span data-testid="badge" className={className}>{children}</span>
+  Badge: ({ children, className, ...rest }: { children?: React.ReactNode; className?: string; [key: string]: unknown }) => (
+    <span data-testid="badge" className={className} {...rest}>{children}</span>
   ),
 }));