diff --git a/src/frontend/tests/extended/features/tool-mode.spec.ts b/src/frontend/tests/extended/features/tool-mode.spec.ts
index 7893801bb..65f132ec3 100644
--- a/src/frontend/tests/extended/features/tool-mode.spec.ts
+++ b/src/frontend/tests/extended/features/tool-mode.spec.ts
@@ -132,7 +132,10 @@ test(
 
     // Move the Agent node a bit
 
-    await page.getByTestId("handle-url-shownode-toolset-right").first().click();
+    await page
+      .getByTestId("handle-urlcomponent-shownode-toolset-right")
+      .first()
+      .click();
 
     await page.getByTestId("handle-agent-shownode-tools-left").first().click();
 
diff --git a/src/frontend/tests/extended/integrations/chatInputOutputUser-shard-1.spec.ts b/src/frontend/tests/extended/integrations/chatInputOutputUser-shard-1.spec.ts
index 922fceecf..3d1bc061c 100644
--- a/src/frontend/tests/extended/integrations/chatInputOutputUser-shard-1.spec.ts
+++ b/src/frontend/tests/extended/integrations/chatInputOutputUser-shard-1.spec.ts
@@ -115,6 +115,8 @@ test(
       .nth(0)
       .click();
 
+    await page.waitForTimeout(1000);
+
     // Run flow and test text output inspection
     await page.getByTestId("button_run_url").first().click();
     await page.waitForSelector("text=built successfully", {
@@ -185,7 +187,6 @@ test(
     const dataEdge = await page.locator(".react-flow__edge").first();
     await dataEdge.click();
     await page.keyboard.press("Backspace");
-    await page.waitForTimeout(600);
 
     await page.waitForTimeout(5000);