diff --git a/crates/ai/src/assistant.rs b/crates/ai/src/assistant.rs
index 2d3c21ba3112620924379ee1e53ef2a60f3bfb63..770aff954d0db45299ee8463b8687334950bceb1 100644
--- a/crates/ai/src/assistant.rs
+++ b/crates/ai/src/assistant.rs
@@ -2244,11 +2244,10 @@ mod tests {
             workspace.add_panel(panel.clone(), cx);
             workspace.toggle_dock(DockPosition::Right, cx);
             assert!(workspace.right_dock().read(cx).is_open());
-            cx.focus(&panel);
         });
 
+        panel.update(cx, |_, cx| cx.focus_self());
         cx.dispatch_action(window_id, workspace::ToggleZoom);
-
         workspace.read_with(cx, |workspace, cx| {
             assert_eq!(workspace.zoomed_view(cx).unwrap(), panel);
         })