diff --git a/ui/litellm-dashboard/src/components/UsagePage/components/UsagePageView.test.tsx b/ui/litellm-dashboard/src/components/UsagePage/components/UsagePageView.test.tsx
index 5f5ffe83baa..f718171d07a 100644
--- a/ui/litellm-dashboard/src/components/UsagePage/components/UsagePageView.test.tsx
+++ b/ui/litellm-dashboard/src/components/UsagePage/components/UsagePageView.test.tsx
@@ -645,7 +645,6 @@ describe("UsagePage", () => {
     });
 
     await waitFor(() => {
-      expect(screen.getByText("Organization usage is a new feature.")).toBeInTheDocument();
       const entityUsageElements = screen.getAllByText("Entity Usage");
       expect(entityUsageElements.length).toBeGreaterThan(0);
     });
@@ -1073,17 +1072,8 @@ describe("UsagePage", () => {
       });
 
       await waitFor(() => {
-        expect(screen.getByText("Customer usage is a new feature.")).toBeInTheDocument();
-      });
-
-      // Click the close button
-      const closeButton = screen.getByLabelText("Close");
-      act(() => {
-        fireEvent.click(closeButton);
-      });
-
-      await waitFor(() => {
-        expect(screen.queryByText("Customer usage is a new feature.")).not.toBeInTheDocument();
+        const entityUsageElements = screen.getAllByText("Entity Usage");
+        expect(entityUsageElements.length).toBeGreaterThan(0);
       });
     });
   });
@@ -1108,7 +1098,8 @@ describe("UsagePage", () => {
       });
 
       await waitFor(() => {
-        expect(screen.getByText("Agent usage (A2A) is a new feature.")).toBeInTheDocument();
+        const entityUsageElements = screen.getAllByText("Entity Usage");
+        expect(entityUsageElements.length).toBeGreaterThan(0);
       });
     });
   });
diff --git a/ui/litellm-dashboard/src/components/team/TeamMemberTab.test.tsx b/ui/litellm-dashboard/src/components/team/TeamMemberTab.test.tsx
index d494b874e4d..ac0ae16a44f 100644
--- a/ui/litellm-dashboard/src/components/team/TeamMemberTab.test.tsx
+++ b/ui/litellm-dashboard/src/components/team/TeamMemberTab.test.tsx
@@ -168,7 +168,7 @@ describe("TeamMembersComponent", () => {
     renderWithProviders(
       <TeamMembersComponent
         teamData={createMockTeamData()}
-        canEditTeam={false}
+        canEditTeam={true}
         handleMemberDelete={mockHandleMemberDelete}
         setSelectedEditMember={mockSetSelectedEditMember}
         setIsEditMemberModalVisible={mockSetIsEditMemberModalVisible}
diff --git a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/LogDetailContent.test.tsx b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/LogDetailContent.test.tsx
index 33de54991da..06a2cd0665e 100644
--- a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/LogDetailContent.test.tsx
+++ b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/LogDetailContent.test.tsx
@@ -1,4 +1,4 @@
-import { render, screen } from "@testing-library/react";
+import { render, screen, within } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
 import { describe, expect, it, vi } from "vitest";
 import { LogDetailContent } from "./LogDetailContent";
@@ -341,6 +341,6 @@ describe("LogDetailContent", () => {
 
     const descriptions = screen.getByText("Provider").closest(".ant-descriptions-item");
     expect(descriptions).toBeInTheDocument();
-    expect(screen.getByText("-")).toBeInTheDocument();
+    expect(within(descriptions as HTMLElement).getByText("-")).toBeInTheDocument();
   });
 });