diff --git a/ui/litellm-dashboard/src/components/model_add/CredentialsTable.test.tsx b/ui/litellm-dashboard/src/components/model_add/CredentialsTable.test.tsx index 47e68d479b8..5fcb62eea61 100644 --- a/ui/litellm-dashboard/src/components/model_add/CredentialsTable.test.tsx +++ b/ui/litellm-dashboard/src/components/model_add/CredentialsTable.test.tsx @@ -67,10 +67,10 @@ describe("CredentialsTable", () => { ]; render(); - const aliasedRow = screen.getByText("aliased").closest("tr")!; + const aliasedRow = screen.getByRole("row", { name: /^aliased/ }); expect(within(aliasedRow).getByText("Prod OpenAI")).toBeInTheDocument(); - const unaliasedRow = screen.getByText("unaliased").closest("tr")!; + const unaliasedRow = screen.getByRole("row", { name: /^unaliased/ }); expect(within(unaliasedRow).getByText("-")).toBeInTheDocument(); });