diff --git a/source/binding/Python/maa/agent/agent_server.py b/source/binding/Python/maa/agent/agent_server.py
index c40160012..3e864439a 100644
--- a/source/binding/Python/maa/agent/agent_server.py
+++ b/source/binding/Python/maa/agent/agent_server.py
@@ -67,7 +67,7 @@ def register_custom_action(name: str, action: "CustomAction") -> bool:  # type:
         )
 
     @staticmethod
-    def start_up(identifier: str = sys.argv[-1]) -> bool:
+    def start_up(identifier: str) -> bool:
 
         AgentServer._set_api_properties()
 
diff --git a/test/agent/agent_child_test.py b/test/agent/agent_child_test.py
index 5de1674e8..8f4cb323a 100644
--- a/test/agent/agent_child_test.py
+++ b/test/agent/agent_child_test.py
@@ -27,8 +27,10 @@
 
 
 def main():
-    AgentServer.start_up(sys.argv[-1])
+    socket_id = sys.argv[-1]
+    AgentServer.start_up(socket_id)
     AgentServer.join()
+    AgentServer.shut_down()
 
 
 @AgentServer.custom_recognition("MyRec")