diff --git a/ui/litellm-dashboard/src/components/entity_usage.test.tsx b/ui/litellm-dashboard/src/components/entity_usage.test.tsx index 2b6234c039e..2c070427ead 100644 --- a/ui/litellm-dashboard/src/components/entity_usage.test.tsx +++ b/ui/litellm-dashboard/src/components/entity_usage.test.tsx @@ -245,9 +245,9 @@ describe("EntityUsage", () => { expect(mockTagDailyActivityCall).toHaveBeenCalled(); }); - expect(await screen.findByText("Tag Spend Overview")).toBeInTheDocument(); - expect(await screen.findByText("$-")).toBeInTheDocument(); - expect(screen.getByText("Total Spend")).toBeInTheDocument(); - expect(screen.getAllByText("0")[0]).toBeInTheDocument(); + await waitFor(() => { + expect(screen.getByText("$0.00")).toBeInTheDocument(); + expect(screen.getByText("Total Spend")).toBeInTheDocument(); + }); }); });