diff --git a/agents-api/agents_api/activities/task_steps/prompt_step.py b/agents-api/agents_api/activities/task_steps/prompt_step.py
index 7f08e8d1..968cff8b 100644
--- a/agents-api/agents_api/activities/task_steps/prompt_step.py
+++ b/agents-api/agents_api/activities/task_steps/prompt_step.py
@@ -164,8 +164,10 @@ async def prompt_step(context: StepContext) -> StepOutcome:
 
         # Filter tools for specific types
         filtered_tools = [
-            tool for tool in formatted_tools
-            if tool["type"] in ["computer_20241022", "bash_20241022", "text_editor_20241022"]
+            tool
+            for tool in formatted_tools
+            if tool["type"]
+            in ["computer_20241022", "bash_20241022", "text_editor_20241022"]
         ]
 
         # Claude Response