diff --git a/ui/src/components/agent-config-primitives.tsx b/ui/src/components/agent-config-primitives.tsx
index e66bf4d8..481aa4b0 100644
--- a/ui/src/components/agent-config-primitives.tsx
+++ b/ui/src/components/agent-config-primitives.tsx
@@ -105,11 +105,13 @@ export function ToggleField({
hint,
checked,
onChange,
+ toggleTestId,
}: {
label: string;
hint?: string;
checked: boolean;
onChange: (v: boolean) => void;
+ toggleTestId?: string;
}) {
return (
@@ -119,6 +121,8 @@ export function ToggleField({