diff --git a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/GuardrailsPanel.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/GuardrailsPanel.test.tsx
index 2ad332ae9ed..6d05dfaba58 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/GuardrailsPanel.test.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/GuardrailsPanel.test.tsx
@@ -48,12 +48,16 @@ vi.mock("@/utils/roles", () => ({
   isAdminRole: vi.fn((role: string) => role === "admin"),
 }));
 
-vi.mock("./guardrail_info_helpers", () => ({
-  getGuardrailLogoAndName: vi.fn(() => ({
-    logo: null,
-    displayName: "Test Provider",
-  })),
-}));
+vi.mock("./guardrail_info_helpers", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("./guardrail_info_helpers")>();
+  return {
+    ...actual,
+    getGuardrailLogoAndName: vi.fn(() => ({
+      logo: null,
+      displayName: "Test Provider",
+    })),
+  };
+});
 
 beforeAll(() => {
   Object.defineProperty(window, "matchMedia", {