From 9a0bc5e491b2f16dde28fced336f112388362928 Mon Sep 17 00:00:00 2001
From: Ahmed Allam <49919286+0xallam@users.noreply.github.com>
Date: Thu, 19 Mar 2026 20:39:05 -0700
Subject: [PATCH] fix: prevent ScreenStackError when stopping agent from modal
 (#374)

---
 strix/interface/tui.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/strix/interface/tui.py b/strix/interface/tui.py
index 7f453ba..7a94b16 100644
--- a/strix/interface/tui.py
+++ b/strix/interface/tui.py
@@ -252,10 +252,9 @@ class StopAgentScreen(ModalScreen):  # type: ignore[misc]
             event.prevent_default()
 
     def on_button_pressed(self, event: Button.Pressed) -> None:
+        self.app.pop_screen()
         if event.button.id == "stop_agent":
             self.app.action_confirm_stop_agent(self.agent_id)
-        else:
-            self.app.pop_screen()
 
 
 class VulnerabilityDetailScreen(ModalScreen):  # type: ignore[misc]
@@ -1912,8 +1911,6 @@ class StrixTUIApp(App):  # type: ignore[misc]
         return agent_name, False
 
     def action_confirm_stop_agent(self, agent_id: str) -> None:
-        self.pop_screen()
-
         try:
             from strix.tools.agents_graph.agents_graph_actions import stop_agent