diff --git a/core/agents/bug_hunter.py b/core/agents/bug_hunter.py
index 430ba208..ffbcbc77 100644
--- a/core/agents/bug_hunter.py
+++ b/core/agents/bug_hunter.py
@@ -105,7 +105,7 @@ class BugHunter(BaseAgent):
             "fix_attempted": False
         }]
 
-        if False and hunt_conclusion.conclusion == magic_words.PROBLEM_IDENTIFIED:
+        if hunt_conclusion.conclusion == magic_words.PROBLEM_IDENTIFIED:
             # if no need for logs, implement iteration same as before
             self.next_state.current_iteration["status"] = IterationStatus.AWAITING_BUG_FIX
             await self.send_message("The bug is found - I'm  attempting to fix it.")
diff --git a/core/agents/troubleshooter.py b/core/agents/troubleshooter.py
index 7410b45c..f39199e0 100644
--- a/core/agents/troubleshooter.py
+++ b/core/agents/troubleshooter.py
@@ -33,7 +33,7 @@ class Troubleshooter(IterationPromptMixin, BaseAgent):
     agent_type = "troubleshooter"
     display_name = "Troubleshooter"
 
-    async def run(self):
+    async def run(self) -> AgentResponse:
         if self.current_state.unfinished_iterations:
             if self.current_state.current_iteration.get("status") == IterationStatus.FIND_SOLUTION:
                 return await self.propose_solution()
@@ -98,7 +98,9 @@ class Troubleshooter(IterationPromptMixin, BaseAgent):
                 await self.trace_loop("loop-feedback")
         elif bug_report is not None:
             iteration_status = IterationStatus.HUNTING_FOR_BUG
-        elif change_description is not None:
+        else:
+            # should be - elif change_description is not None: - but to prevent bugs with the extension
+            # this might be caused if we show the input field instead of buttons
             iteration_status = IterationStatus.FIND_SOLUTION