1 files changed, 8 insertions, 7 deletions

diff --git a/recipes/amazon-work-mail/webview.js b/recipes/amazon-work-mail/webview.js
index 944a15c..44f3058 100644
--- a/recipes/amazon-work-mail/webview.js
+++ b/recipes/amazon-work-mail/webview.js
@@ -1,6 +1,6 @@
-module.exports = (Ferdi) => {
-  const reload = (EventType) => new Promise((resolve, reject) => {
-    const btn = document.getElementsByClassName('giraffe-hierarchy-node-refresh')[0];
+const reload = EventType =>
+  new Promise((resolve, reject) => {
+    const btn = document.querySelectorAll('.giraffe-hierarchy-node-refresh')[0];
     const EventObject = document.createEvent('Events');
     EventObject.initEvent(EventType, true, false);
 
@@ -11,12 +11,13 @@ module.exports = (Ferdi) => {
     }
   });
 
+module.exports = Ferdi => {
   const getUnread = () => {
-    const nodes = document.getElementsByClassName('giraffe-hierarchy-node-counter');
+    const nodes = document.querySelectorAll('.giraffe-hierarchy-node-counter');
     let counter = 0;
 
-    for (let i = 0; i < nodes.length; i++) {
-      counter += Ferdi.safeParseInt(nodes[i].innerText);
+    for (const node of nodes) {
+      counter += Ferdi.safeParseInt(node.textContent);
     }
 
     Ferdi.setBadge(counter);
@@ -27,6 +28,6 @@ module.exports = (Ferdi) => {
 
     window.setInterval(() => {
       reload('click');
-    }, 60000);
+    }, 60_000);
   }
 };