diff --git a/agent/python/agent_protocol/agent.py b/agent/python/agent_protocol/agent.py
index 31efc84..a3440d7 100644
--- a/agent/python/agent_protocol/agent.py
+++ b/agent/python/agent_protocol/agent.py
@@ -136,7 +136,7 @@ async def list_agent_task_artifacts(task_id: str) -> List[Artifact]:
     "/agent/tasks/{task_id}/artifacts/{artifact_id}",
     tags=["agent"],
 )
-async def get_agent_task_artifacts(task_id: str, artifact_id: str) -> FileResponse:
+async def download_agent_task_artifacts(task_id: str, artifact_id: str) -> FileResponse:
     """
     Download the specified artifact.
     """
diff --git a/agent/python/pyproject.toml b/agent/python/pyproject.toml
index a6d2513..2e6e8af 100644
--- a/agent/python/pyproject.toml
+++ b/agent/python/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agent-protocol"
-version = "0.2.1"
+version = "0.2.2"
 description = "API for interacting with Agent"
 authors = ["e2b <hello@e2b.dev>"]
 license = "MIT"
diff --git a/openapi.yml b/openapi.yml
index 7b8afd9..907791d 100644
--- a/openapi.yml
+++ b/openapi.yml
@@ -147,7 +147,7 @@ paths:
     get:
       summary: List all artifacts that have been created for the given task.
       tags: [agent]
-      operationId: getAgentTaskArtifact
+      operationId: listAgentTaskArtifacts
       parameters:
         - name: task_id
           in: path
@@ -155,12 +155,6 @@ paths:
           required: true
           schema:
             type: string
-        - name: artifact_id
-          in: path
-          description: ID of the artifact
-          required: true
-          schema:
-            type: string
       responses:
         200:
           description: Returned the content of the artifact.