diff --git a/ui/litellm-dashboard/src/components/common_components/KeyLifecycleSettings.test.tsx b/ui/litellm-dashboard/src/components/common_components/KeyLifecycleSettings.test.tsx
index 462b0a3d99d..c00e8008358 100644
--- a/ui/litellm-dashboard/src/components/common_components/KeyLifecycleSettings.test.tsx
+++ b/ui/litellm-dashboard/src/components/common_components/KeyLifecycleSettings.test.tsx
@@ -53,7 +53,7 @@ vi.mock("@tremor/react", () => ({
     };
     return (
       <input
-        data-testid={name === "duration" ? "duration-input" : "custom-interval-input"}
+        data-testid={name === "duration" ? "duration-input" : name === "key_rotation_email" ? "key-rotation-email-input" : "custom-interval-input"}
         value={value}
         onChange={handleChange}
         placeholder={placeholder}