diff --git a/ui/litellm-dashboard/src/components/AIHub/AgentHubTableColumns.test.tsx b/ui/litellm-dashboard/src/components/AIHub/AgentHubTableColumns.test.tsx
index 703a24d8d6..083e67c297 100644
--- a/ui/litellm-dashboard/src/components/AIHub/AgentHubTableColumns.test.tsx
+++ b/ui/litellm-dashboard/src/components/AIHub/AgentHubTableColumns.test.tsx
@@ -22,9 +22,17 @@ const mockAgent: AgentHubData = {
   is_public: true,
 };
 
-function TestTable({ data, publicPage = false }: { data: AgentHubData[]; publicPage?: boolean }) {
-  const showModal = vi.fn();
-  const copyToClipboard = vi.fn();
+function TestTable({
+  data,
+  publicPage = false,
+  showModal = vi.fn(),
+  copyToClipboard = vi.fn(),
+}: {
+  data: AgentHubData[];
+  publicPage?: boolean;
+  showModal?: ReturnType<typeof vi.fn>;
+  copyToClipboard?: ReturnType<typeof vi.fn>;
+}) {
   const columns = getAgentHubTableColumns(showModal, copyToClipboard, publicPage);
   const table = useReactTable({ data, columns, getCoreRowModel: getCoreRowModel() });
 
diff --git a/ui/litellm-dashboard/src/components/EntityUsageExport/ExportFormatSelector.test.tsx b/ui/litellm-dashboard/src/components/EntityUsageExport/ExportFormatSelector.test.tsx
index a88f9cb116..d20d24992f 100644
--- a/ui/litellm-dashboard/src/components/EntityUsageExport/ExportFormatSelector.test.tsx
+++ b/ui/litellm-dashboard/src/components/EntityUsageExport/ExportFormatSelector.test.tsx
@@ -27,9 +27,7 @@ describe("ExportFormatSelector", () => {
     // Open the Ant Design Select dropdown
     await user.click(screen.getByText("CSV (Excel, Google Sheets)"));
     // Select JSON option from the dropdown
-    const jsonOption = await screen.findByText("JSON (includes metadata)", {
-      selector: ".ant-select-item-option-content",
-    });
+    const jsonOption = await screen.findByText("JSON (includes metadata)");
     await user.click(jsonOption);
     expect(onChange).toHaveBeenCalledWith("json", expect.anything());
   });