diff --git a/ui/src/ui/controllers/agents.test.ts b/ui/src/ui/controllers/agents.test.ts
index 8dacdbb20ed..a026d447cf9 100644
--- a/ui/src/ui/controllers/agents.test.ts
+++ b/ui/src/ui/controllers/agents.test.ts
@@ -28,13 +28,27 @@ function createSaveState(): {
   return {
     state: {
       ...state,
+      applySessionKey: "session-1",
+      configLoading: false,
+      configRawOriginal: "{}",
+      configValid: true,
+      configIssues: [],
       configSaving: false,
+      configApplying: false,
+      updateRunning: false,
       configSnapshot: { hash: "hash-1" },
       configFormDirty: true,
       configFormMode: "form",
       configForm: { agents: { list: [{ id: "main" }] } },
       configRaw: "{}",
       configSchema: null,
+      configSchemaVersion: null,
+      configSchemaLoading: false,
+      configUiHints: {},
+      configFormOriginal: { agents: { list: [{ id: "main" }] } },
+      configSearchQuery: "",
+      configActiveSection: null,
+      configActiveSubsection: null,
       lastError: null,
     },
     request,