diff --git a/e2e/client/specs/helpers/tree-select-driver.ts b/e2e/client/specs/helpers/tree-select-driver.ts
index c2f4162070..aeeac91729 100644
--- a/e2e/client/specs/helpers/tree-select-driver.ts
+++ b/e2e/client/specs/helpers/tree-select-driver.ts
@@ -25,9 +25,7 @@ export class TreeSelectDriver {
     addValue(value: string): void {
         el(['open-popover'], null, this._element).click();
 
-        const dropdown = element(by.id('TREESELECT_DROPDOWN'));
-
-        dropdown.element(by.cssContainingText('[data-test-id="option"]', value)).click();
+        element(by.cssContainingText('[data-test-id="tree-select-popover"] [data-test-id="option"]', value)).click();
     }
 
     setValue(value: string) {