diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py
index 9a551108..ed2e2435 100644
--- a/skyvern/forge/sdk/db/client.py
+++ b/skyvern/forge/sdk/db/client.py
@@ -332,6 +332,7 @@ class AgentDB:
                 ).first():
                     task.failure_reason = None
                     await session.commit()
+                    await session.refresh(task)
                     return convert_to_task(task, debug_enabled=self.debug_enabled)
                 else:
                     raise NotFoundError("Task not found")