```diff
--- test_files/118-original.txt	2025-03-07 19:06:23
+++ test_files/118-modified.txt	2025-03-07 19:06:23
@@ -25,7 +25,7 @@
 )
 composio_toolset = ComposioToolSet(
     workspace_config=WorkspaceType.Docker(
-        image="composio/composio:dev", persistent=True
+        image="composio/composio:latest", persistent=True
     )
 )
 
@@ -109,7 +109,7 @@
 # Router function
 def router(
     state,
-) -> Literal["call_tool", "pm", "__end__", "continue",]:
+) -> Literal["call_tool", "pm", "__end__", "continue"]:
     last_message = state["messages"][-1]
     sender = state["sender"]
     if last_message.tool_calls:
```
