diff --git a/ui/litellm-dashboard/src/components/playground/compareUI/components/ComparisonPanel.test.tsx b/ui/litellm-dashboard/src/components/playground/compareUI/components/ComparisonPanel.test.tsx
index 8a120e86485..2aef47f71d9 100644
--- a/ui/litellm-dashboard/src/components/playground/compareUI/components/ComparisonPanel.test.tsx
+++ b/ui/litellm-dashboard/src/components/playground/compareUI/components/ComparisonPanel.test.tsx
@@ -3,15 +3,16 @@ import userEvent from "@testing-library/user-event";
import { beforeEach, describe, expect, it, vi } from "vitest";
import type { ComparisonInstance } from "../CompareUI";
import { ComparisonPanel } from "./ComparisonPanel";
+import { EndpointId, ENDPOINT_CONFIGS } from "../endpoint_config";
vi.mock("./MessageDisplay", () => ({
MessageDisplay: () =>
MessageDisplay
,
}));
-vi.mock("./ModelSelector", () => ({
- ModelSelector: ({ value, onChange }: { value: string; onChange: (val: string) => void }) => (
-