diff --git a/framework/Element.js b/framework/Element.js
index c763d00..2411c0e 100644
--- a/framework/Element.js
+++ b/framework/Element.js
@@ -28,7 +28,7 @@ export default class Element {
 
     async waitUntilVisible(timeout = defaultTimeout) {
         console.log(`Waiting for ${this.selector} to be visible...`);
-        const elementHandle = await page.waitForSelector(this.selector, { visible: true, timeout: timeout });
+        const elementHandle = await page.waitForSelector(this.selector, { state: "visible", timeout: timeout });
         if (config.captureScreenshots) {
             await Helpers.takeScreenshot();
         }
@@ -37,7 +37,7 @@ export default class Element {
 
     async waitUntilInvisible(timeout = defaultTimeout) {
         console.log(`Waiting for ${this.selector} to be invisible...`);
-        await page.waitForSelector(this.selector, { hidden: true, timeout: timeout });
+        await page.waitForSelector(this.selector, { state: "hidden", timeout: timeout });
         if (config.captureScreenshots) {
             await Helpers.takeScreenshot();
         }