diff --git a/integrations/langchain/pyproject.toml b/integrations/langchain/pyproject.toml
index d529db83..dc0bc6e4 100644
--- a/integrations/langchain/pyproject.toml
+++ b/integrations/langchain/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "skyvern-langchain"
-version = "0.1.5"
+version = "0.1.6"
 description = ""
 authors = ["lawyzheng <lawy@skyvern.com>"]
 packages = [{ include = "skyvern_langchain" }]
diff --git a/integrations/langchain/skyvern_langchain/__init__.py b/integrations/langchain/skyvern_langchain/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/integrations/langchain/skyvern_langchain/agent.py b/integrations/langchain/skyvern_langchain/agent.py
index 15e26ddc..f9d2cd31 100644
--- a/integrations/langchain/skyvern_langchain/agent.py
+++ b/integrations/langchain/skyvern_langchain/agent.py
@@ -18,7 +18,7 @@ agent = Agent()
 
 class SkyvernTaskBaseTool(BaseTool):
     engine: Literal["TaskV1", "TaskV2"] = Field(default=settings.engine)
-    timeout_seconds: int = Field(default=settings.run_task_timeout)
+    run_task_timeout_seconds: int = Field(default=settings.run_task_timeout_seconds)
     agent: Agent = agent
 
     def _run(self, *args: Any, **kwargs: Any) -> None:
@@ -48,11 +48,13 @@ class RunTask(SkyvernTaskBaseTool):
         if url is not None:
             task_request.url = url
 
-        return await self.agent.run_task(task_request=task_request, timeout_seconds=self.timeout_seconds)
+        return await self.agent.run_task(task_request=task_request, timeout_seconds=self.run_task_timeout_seconds)
 
     async def _arun_task_v2(self, user_prompt: str, url: str | None = None) -> ObserverTask:
         task_request = ObserverTaskRequest(user_prompt=user_prompt, url=url)
-        return await self.agent.run_observer_task_v_2(task_request=task_request, timeout_seconds=self.timeout_seconds)
+        return await self.agent.run_observer_task_v_2(
+            task_request=task_request, timeout_seconds=self.run_task_timeout_seconds
+        )
 
 
 class DispatchTask(SkyvernTaskBaseTool):
diff --git a/integrations/langchain/skyvern_langchain/client.py b/integrations/langchain/skyvern_langchain/client.py
index 303d71c2..7ba6c18f 100644
--- a/integrations/langchain/skyvern_langchain/client.py
+++ b/integrations/langchain/skyvern_langchain/client.py
@@ -15,7 +15,7 @@ class SkyvernTaskBaseTool(BaseTool):
     api_key: str = Field(default=settings.api_key)
     base_url: str = Field(default=settings.base_url)
     engine: Literal["TaskV1", "TaskV2"] = Field(default=settings.engine)
-    run_task_timeout_seconds: int = Field(default=settings.run_task_timeout)
+    run_task_timeout_seconds: int = Field(default=settings.run_task_timeout_seconds)
 
     def get_client(self) -> AsyncSkyvern:
         httpx_client = AsyncClient(
@@ -46,10 +46,10 @@ class RunTask(SkyvernTaskBaseTool):
             prompt=user_prompt,
         )
 
-        if url is not None:
-            task_generation.url = url
-
         task_request = TaskRequest.model_validate(task_generation, from_attributes=True)
+        if url is not None:
+            task_request.url = url
+
         return await self.get_client().agent.run_task(
             timeout_seconds=self.run_task_timeout_seconds,
             url=task_request.url,
@@ -89,10 +89,10 @@ class DispatchTask(SkyvernTaskBaseTool):
             prompt=user_prompt,
         )
 
-        if url is not None:
-            task_generation.url = url
-
         task_request = TaskRequest.model_validate(task_generation, from_attributes=True)
+        if url is not None:
+            task_request.url = url
+
         return await self.get_client().agent.create_task(
             url=task_request.url,
             title=task_request.title,