diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py
index 0b0422be..259c235f 100644
--- a/skyvern/forge/sdk/db/client.py
+++ b/skyvern/forge/sdk/db/client.py
@@ -322,7 +322,7 @@ class AgentDB:
             LOG.error("UnexpectedError", exc_info=True)
             raise
 
-    async def clear_task_failure_reason(self, task_id: str, organization_id: str) -> Task:
+    async def clear_task_failure_reason(self, organization_id: str, task_id: str) -> Task:
         try:
             async with self.Session() as session:
                 if task := (