diff --git a/tests/browser/01_io.js b/tests/browser/01_io.js
index f232bc10..09024ed0 100644
--- a/tests/browser/01_io.js
+++ b/tests/browser/01_io.js
@@ -173,9 +173,15 @@ module.exports = {
 
         browser.waitForElementVisible("#stale-indicator");
 
+        browser.expect.element("#auto-bake").to.not.be.selected;
+
         // Enable previously disabled autobake
         browser.click("#auto-bake-label");
 
+        browser.waitUntil(() => {
+            return browser.expect.element("#auto-bake").to.be.selected;
+        }, 1000);
+
         browser.sendKeys("#input-text .cm-content", "1");
 
         browser.waitForElementVisible("#output-loader");