diff --git a/ui/src/components/agent-config-primitives.tsx b/ui/src/components/agent-config-primitives.tsx
index 70694f73c1c..6eafd947d1f 100644
--- a/ui/src/components/agent-config-primitives.tsx
+++ b/ui/src/components/agent-config-primitives.tsx
@@ -104,11 +104,13 @@ export function ToggleField({
hint,
checked,
onChange,
+ toggleTestId,
}: {
label: string;
hint?: string;
checked: boolean;
onChange: (v: boolean) => void;
+ toggleTestId?: string;
}) {
return (
@@ -118,6 +120,8 @@ export function ToggleField({