diff --git a/ui/litellm-dashboard/src/components/edit_auto_router/auto_router_tier_model_replacement.test.ts b/ui/litellm-dashboard/src/components/edit_auto_router/auto_router_tier_model_replacement.test.ts index 68df4f6b11b..374c81a021e 100644 --- a/ui/litellm-dashboard/src/components/edit_auto_router/auto_router_tier_model_replacement.test.ts +++ b/ui/litellm-dashboard/src/components/edit_auto_router/auto_router_tier_model_replacement.test.ts @@ -30,13 +30,14 @@ describe("auto-router tier model replacement", () => { models: ["gemma4-12b-it-optiq-4bit"], }).value; const saved = buildUpdatedComplexityRouterConfig(stored, changed); - - expect(saved.tiers).toEqual({ + const expectedTiers = { SIMPLE: ["gemma4-12b-it-optiq-4bit"], MEDIUM: [], COMPLEX: [], REASONING: [], - }); + }; + + expect(saved.tiers).toEqual(expectedTiers); expect(saved).not.toHaveProperty("tier_model_configs"); }); });