diff --git a/src/frontend/tests/core/unit/codeAreaModalComponent.spec.ts b/src/frontend/tests/core/unit/codeAreaModalComponent.spec.ts
index 47a8e1339..398aecd3b 100644
--- a/src/frontend/tests/core/unit/codeAreaModalComponent.spec.ts
+++ b/src/frontend/tests/core/unit/codeAreaModalComponent.spec.ts
@@ -52,10 +52,11 @@ test(
     await page.waitForSelector('[data-testid="prototypesPython Function"]', {
       timeout: 3000,
     });
-
+    await page.getByTestId("prototypesPython Function").hover();
     await page
       .getByTestId("prototypesPython Function")
-      .dragTo(page.locator('//*[@id="react-flow-id"]'));
+      .getByTestId("icon-Plus")
+      .click();
     await page.getByTestId("fit_view").click();
     await page.getByTestId("zoom_out").click();
     await page.getByTestId("div-generic-node").click();