diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py
index 59ec7a2d..7f660fb3 100644
--- a/vision_agent/agent/vision_agent_v2.py
+++ b/vision_agent/agent/vision_agent_v2.py
@@ -93,12 +93,7 @@ def write_code(
 
 
 def write_test(
-    user_req: str,
-    subtask: str,
-    tool_info: str,
-    _: str,
-    code: str,
-    model: LLM
+    user_req: str, subtask: str, tool_info: str, _: str, code: str, model: LLM
 ) -> str:
     prompt = TEST.format(
         context=USER_REQ_SUBTASK_CONTEXT.format(