diff --git a/tests/helpers/docsifyInit.js b/tests/helpers/docsifyInit.js
index f3bf6479..68d24e24 100644
--- a/tests/helpers/docsifyInit.js
+++ b/tests/helpers/docsifyInit.js
@@ -14,6 +14,8 @@ async function docsifyInit(page, options) {
     script: '',
     scriptURLs: [],
     url: `http://${serverIP}:${serverPort}`,
+    docsifyURL: '/lib/docsify.js',
+    waitForSelector: '#main',
   };
   const settings = {
     ...defaults,
@@ -73,6 +75,11 @@ async function docsifyInit(page, options) {
   for (const url of settings.scriptURLs) {
     await page.addScriptTag({ url });
   }
+
+  // Detect docsify "complete" by waiting for specified element
+  if (settings.waitForSelector) {
+    await page.waitForSelector(settings.waitForSelector);
+  }
 }
 
 module.exports = docsifyInit;