1 files changed, 6 insertions, 6 deletions

diff --git a/recipes/erepublik/webview.js b/recipes/erepublik/webview.js
index 6a105e1..a47c57d 100644
--- a/recipes/erepublik/webview.js
+++ b/recipes/erepublik/webview.js
@@ -4,19 +4,19 @@ function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { de
 
 module.exports = Ferdi => {
   const getMessages = () => {
-    const elementNotify = document.getElementsByClassName('notify');
-    const elementFeed = document.getElementsByClassName('unreadCounter ng-binding ng-scope');
+    const elementNotify = document.querySelectorAll('.notify');
+    const elementFeed = document.querySelectorAll('.unreadCounter.ng-binding.ng-scope');
 
     let countNotify = 0;
     let countFeed = 0;
 
-    for (let i = 0; i < elementNotify.length; i++) {
-      const splitText = elementNotify[i].title.split(':');
+    for (const element of elementNotify) {
+      const splitText = element.title.split(':');
       countNotify += Ferdi.safeParseInt(splitText[1]);
     }
 
-    for (let i = 0; i < elementFeed.length; i++) {
-      countFeed += Ferdi.safeParseInt(elementFeed[i].textContent);
+    for (const element of elementFeed) {
+      countFeed += Ferdi.safeParseInt(element.textContent);
     }
 
     Ferdi.setBadge(countNotify, countFeed);