diff --git a/apps/remix-ide-e2e/src/tests/sol2uml.test.ts b/apps/remix-ide-e2e/src/tests/sol2uml.test.ts
index 4f29da084d..a0e1eddb7e 100644
--- a/apps/remix-ide-e2e/src/tests/sol2uml.test.ts
+++ b/apps/remix-ide-e2e/src/tests/sol2uml.test.ts
@@ -31,7 +31,7 @@ module.exports = {
             .rightClick('*[data-id="treeViewLitreeViewItemsecondContract.sol"]')
             .click('*[id="menuitemgeneratecustomaction"')
             .waitForElementVisible('*[id="sol-uml-gen"]')
-            .click('*[id="umlZoominbtn"]')
+            .click('*[data-id="umlZoominbtn"]')
     }
 }
 
@@ -215,4 +215,4 @@ contract MyToken is Initializable, ERC721Upgradeable, OwnableUpgradeable, UUPSUp
 }
 `}
 }
-]
\ No newline at end of file
+]