diff --git a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.integration.test.tsx
similarity index 95%
rename from ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.test.tsx
rename to ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.integration.test.tsx
index b0f0265f067..969844c58ed 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.test.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.integration.test.tsx
@@ -33,13 +33,13 @@ describe("PiiConfiguration", () => {
     expect(screen.getByText("NO FODSELSNUMMER")).toBeInTheDocument();
   });
 
-  it("adds a normalized custom entity with the first action", () => {
+  it("adds a normalized custom entity with MASK", () => {
     const onEntitySelect = vi.fn();
     const onActionSelect = vi.fn();
     render(
       <PiiConfiguration
         entities={["PERSON"]}
-        actions={["MASK", "BLOCK"]}
+        actions={["BLOCK", "MASK"]}
         selectedEntities={[]}
         selectedActions={{}}
         onEntitySelect={onEntitySelect}
diff --git a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.tsx b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.tsx
index 4efb795bb6a..e6efdf53670 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.tsx
+++ b/ui/litellm-dashboard/src/app/(dashboard)/guardrails/_components/pii_configuration.tsx
@@ -78,7 +78,7 @@ const PiiConfiguration: React.FC<PiiConfigurationProps> = ({
     }
 
     onEntitySelect(normalizedEntityName);
-    onActionSelect(normalizedEntityName, actions[0] ?? "MASK");
+    onActionSelect(normalizedEntityName, "MASK");
   };
 
   return (