1 files changed, 2 insertions, 10 deletions

diff --git a/recipes/slite/webview.js b/recipes/slite/webview.js
index d892c3c..a044399 100644
--- a/recipes/slite/webview.js
+++ b/recipes/slite/webview.js
@@ -1,15 +1,7 @@
-const getInnerInt = selector => {
-  const element = document.querySelector(selector);
-  return element && parseInt(element.innerText);
-};
-
 module.exports = Ferdi => {
   const getMessages = function getMessages() {
-    const direct = (
-      getInnerInt("#app button[data-test-id='notificationsCount']") || 0
-    );
-
-    Ferdi.setBadge(direct);
+    const element = document.querySelector("#app button[data-test-id='notificationsCount']");
+    Ferdi.setBadge(element ? Ferdi.safeParseInt(element.innerText) : 0);
   };
 
   Ferdi.loop(getMessages);