diff --git a/ui/litellm-dashboard/src/app/(dashboard)/search-tools/_components/SearchToolTester.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/search-tools/_components/SearchToolTester.test.tsx
index 3d4144a74e6..196eb6c88b0 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/search-tools/_components/SearchToolTester.test.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/search-tools/_components/SearchToolTester.test.tsx
@@ -123,6 +123,8 @@ describe("SearchToolTester", () => {
     const searchButton = screen.getByRole("button", { name: /search/i });
     await user.click(searchButton);
     expect(screen.getByText("Searching...")).toBeInTheDocument();
+
+    expect(await screen.findByText("Test Result 1")).toBeInTheDocument();
   });
 
   it("should display search results after successful search", async () => {
@@ -407,6 +409,8 @@ describe("SearchToolTester", () => {
     await user.click(searchButton);
     expect(input).toBeDisabled();
     expect(searchButton).toBeDisabled();
+
+    await waitFor(() => expect(searchButton).toBeEnabled());
   });
 
   it("should display result links that open in new tab", async () => {