diff --git a/packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx b/packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx
index 6be2424194..6bfd238b06 100644
--- a/packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx
+++ b/packages/cli/src/ui/layouts/DefaultAppLayout.test.tsx
@@ -38,6 +38,13 @@ const mockUIState = {
   availableTerminalHeight: 20,
   activePtyId: null,
   isBackgroundTaskVisible: true,
+  btwState: {
+    isActive: false,
+    query: '',
+    response: '',
+    isStreaming: false,
+    error: null,
+  },
 } as unknown as UIState;
 
 vi.mock('../contexts/UIStateContext.js', () => ({