diff --git a/skyvern/agent/client.py b/skyvern/agent/client.py
index 393ce2de..30057f94 100644
--- a/skyvern/agent/client.py
+++ b/skyvern/agent/client.py
@@ -81,7 +81,7 @@ class SkyvernClient:
         run_id: str,
     ) -> RunResponse:
         run_obj = await self.client.agent.get_run(run_id=run_id)
-        if run_obj.run_type in [RunType.task_v1, RunType.task_v2]:
+        if run_obj.run_type in [RunType.task_v1, RunType.task_v2, RunType.openai_cua]:
             return TaskRunResponse.model_validate(run_obj.dict())
         elif run_obj.run_type == RunType.workflow_run:
             return WorkflowRunResponse.model_validate(run_obj.dict())