diff --git a/src/drivers/webextension/js/driver.js b/src/drivers/webextension/js/driver.js
index 4568e30b6..739d00275 100644
--- a/src/drivers/webextension/js/driver.js
+++ b/src/drivers/webextension/js/driver.js
@@ -198,6 +198,8 @@ browser.webRequest.onCompleted.addListener(request => {
   }
 });
 
+wappalyzer.driver.document = document;
+
 /**
  * Log messages to console
  */