diff --git a/skyvern/schemas/runs.py b/skyvern/schemas/runs.py
index 4492006a..b4f3ddf5 100644
--- a/skyvern/schemas/runs.py
+++ b/skyvern/schemas/runs.py
@@ -217,8 +217,8 @@ class BaseRunResponse(BaseModel):
 
 
 class TaskRunResponse(BaseRunResponse):
-    run_type: Literal[RunType.task_v1, RunType.task_v2] = Field(
-        description="Type of task run - either task_v1 or task_v2"
+    run_type: Literal[RunType.task_v1, RunType.task_v2, RunType.openai_cua] = Field(
+        description="Types of a task run - task_v1, task_v2, openai_cua"
     )
     run_request: TaskRunRequest | None = Field(
         default=None, description="The original request parameters used to start this task run"