diff --git a/metagpt/actions/design_api.py b/metagpt/actions/design_api.py
index 1447eacc3..55213a4b0 100644
--- a/metagpt/actions/design_api.py
+++ b/metagpt/actions/design_api.py
@@ -135,7 +135,8 @@ class WriteDesign(Action):
         self._save_prd(docs_path, resources_path, context[-1].content)
         self._save_system_design(docs_path, resources_path, content)
 
-    async def run(self, context):
+    async def run(self, *args, **kwargs):
+        context = args[0]
         prompt = PROMPT_TEMPLATE.format(context=context, format_example=FORMAT_EXAMPLE)
         # system_design = await self._aask(prompt)
         system_design = await self._aask_v1(prompt, "system_design", OUTPUT_MAPPING)
diff --git a/metagpt/actions/project_management.py b/metagpt/actions/project_management.py
index 89c59dcda..394f279e8 100644
--- a/metagpt/actions/project_management.py
+++ b/metagpt/actions/project_management.py
@@ -115,7 +115,8 @@ class WriteTasks(Action):
         requirements_path = WORKSPACE_ROOT / ws_name / 'requirements.txt'
         requirements_path.write_text(rsp.instruct_content.dict().get("Required Python third-party packages").strip('"\n'))
 
-    async def run(self, context):
+    async def run(self, *args, **kwargs):
+        context = args[0]
         prompt = PROMPT_TEMPLATE.format(context=context, format_example=FORMAT_EXAMPLE)
         rsp = await self._aask_v1(prompt, "task", OUTPUT_MAPPING)
         self._save(context, rsp)