diff --git a/pyproject.toml b/pyproject.toml
index 1269fbd0..15e07618 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "skyvern"
-version = "0.1.83"
+version = "0.1.84"
 description = ""
 authors = ["Skyvern AI <info@skyvern.com>"]
 readme = "README.md"
diff --git a/skyvern/library/skyvern.py b/skyvern/library/skyvern.py
index 11cc5fe5..a5b32ea0 100644
--- a/skyvern/library/skyvern.py
+++ b/skyvern/library/skyvern.py
@@ -24,7 +24,7 @@ from skyvern.forge.sdk.schemas.tasks import CreateTaskResponse, Task, TaskReques
 from skyvern.forge.sdk.services.org_auth_token_service import API_KEY_LIFETIME
 from skyvern.forge.sdk.workflow.models.workflow import WorkflowRunStatus
 from skyvern.library.constants import DEFAULT_AGENT_HEARTBEAT_INTERVAL, DEFAULT_AGENT_TIMEOUT
-from skyvern.schemas.runs import CUA_ENGINES, ProxyLocation, RunEngine, RunType
+from skyvern.schemas.runs import CUA_ENGINES, ProxyLocation, RunEngine, RunStatus, RunType
 from skyvern.services import run_service, task_v1_service, task_v2_service
 from skyvern.utils import migrate_db
 
@@ -401,7 +401,7 @@ class Skyvern(AsyncSkyvern):
             async with asyncio.timeout(timeout):
                 while True:
                     task_run = await self.agent.get_run(task_run.run_id)
-                    if task_run.status.is_final():
+                    if RunStatus(task_run.status).is_final():
                         break
                     await asyncio.sleep(DEFAULT_AGENT_HEARTBEAT_INTERVAL)
         return TaskRunResponse.model_validate(task_run.dict())