diff --git a/src/frontend/tests/core/regression/generalBugs-shard-9.spec.ts b/src/frontend/tests/core/regression/generalBugs-shard-9.spec.ts
index a41dfa113..ffdcaa648 100644
--- a/src/frontend/tests/core/regression/generalBugs-shard-9.spec.ts
+++ b/src/frontend/tests/core/regression/generalBugs-shard-9.spec.ts
@@ -95,7 +95,7 @@ AI:
 
     //connection 1
     const elementChatMemoryOutput = await page
-      .getByTestId("handle-memory-shownode-text-right")
+      .getByTestId("handle-memory-shownode-message-right")
       .first();
     await elementChatMemoryOutput.hover();
     await page.mouse.down();
diff --git a/src/frontend/tests/extended/regression/generalBugs-shard-3.spec.ts b/src/frontend/tests/extended/regression/generalBugs-shard-3.spec.ts
index 57b35d8f1..378e3e516 100644
--- a/src/frontend/tests/extended/regression/generalBugs-shard-3.spec.ts
+++ b/src/frontend/tests/extended/regression/generalBugs-shard-3.spec.ts
@@ -109,7 +109,7 @@ test(
     await page.mouse.up();
 
     const elementsOpenAiOutput = await page
-      .locator('[data-testid="handle-openaimodel-shownode-text-right"]')
+      .locator('[data-testid="handle-openaimodel-shownode-message-right"]')
       .all();
 
     for (const element of elementsOpenAiOutput) {