diff --git a/src/frontend/tests/end-to-end/twoEdges.spec.ts b/src/frontend/tests/end-to-end/twoEdges.spec.ts
index c727a03c8..ad83c39fd 100644
--- a/src/frontend/tests/end-to-end/twoEdges.spec.ts
+++ b/src/frontend/tests/end-to-end/twoEdges.spec.ts
@@ -25,13 +25,17 @@ test("user should be able to see multiple edges and interact with them", async (
 
   await page.getByText("Vector Store RAG", { exact: true }).last().click();
   await page.waitForTimeout(3000);
-  await page.getByText("Vector Store", { exact: true }).first().isVisible();
-  await page.getByText("Base Retriever", { exact: true }).first().isVisible();
+  await page.getByText("Retriever", { exact: true }).first().isVisible();
   await page.getByText("Search Results", { exact: true }).first().isVisible();
-  await page.getByTestId("icon-Eye").nth(24).click();
-  await page.getByTestId("icon-Eye").nth(25).click();
-  await page.getByTestId("icon-Eye").nth(26).click();
-  await page.getByText("Vector Store", { exact: true }).first().isHidden();
-  await page.getByText("Base Retriever", { exact: true }).first().isHidden();
+  await page.getByTestId("icon-Eye").nth(19).click();
+  await page.getByTestId("icon-Eye").nth(20).click();
+  await page.getByText("Retriever", { exact: true }).first().isHidden();
   await page.getByText("Search Results", { exact: true }).first().isHidden();
+  await page.getByTestId("icon-ChevronDown").last().isVisible();
+  await page.getByTestId("icon-ChevronDown").last().click();
+  await page.getByText("Retriever", { exact: true }).first().isVisible();
+  await page.getByText("Search Results", { exact: true }).first().isVisible();
+
+  await page.getByTestId("icon-EyeOff").nth(0).isVisible();
+  await page.getByTestId("icon-EyeOff").nth(1).isVisible();
 });