diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
index 411e8402e11..7231c126a63 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
@@ -118,6 +118,7 @@ describe("useModelsInfo", () => {
       // exclude_auto_routers defaults off: only the Models + Endpoints table opts in, so
       // every other consumer of this hook keeps seeing auto-routers.
       false,
+      undefined,
     );
     expect(modelInfoCall).toHaveBeenCalledTimes(1);
   });
@@ -145,6 +146,7 @@ describe("useModelsInfo", () => {
       // exclude_auto_routers defaults off: only the Models + Endpoints table opts in, so
       // every other consumer of this hook keeps seeing auto-routers.
       false,
+      undefined,
     );
   });
 
diff --git a/ui/litellm-dashboard/src/components/shared/table_cells/status_badge.test.tsx b/ui/litellm-dashboard/src/components/shared/table_cells/status_badge.test.tsx
index 9eddbff1950..916a637f31b 100644
--- a/ui/litellm-dashboard/src/components/shared/table_cells/status_badge.test.tsx
+++ b/ui/litellm-dashboard/src/components/shared/table_cells/status_badge.test.tsx
@@ -48,7 +48,7 @@ describe("StatusBadge", () => {
     render(<StatusBadge tone="info" label="gpt-4.1" href="/models-and-endpoints?model_group=gpt-4.1" />);
     const link = screen.getByRole("link", { name: "gpt-4.1" });
     expect(link).toHaveAttribute("href", "/models-and-endpoints?model_group=gpt-4.1");
-    expect(link.className).toContain("text-info");
+    expect(link).toHaveClass("text-info");
     await user.click(link);
     expect(push).toHaveBeenCalledWith("/models-and-endpoints?model_group=gpt-4.1");
   });