1 files changed, 11 insertions, 5 deletions

diff --git a/recipes/producthunt/webview.js b/recipes/producthunt/webview.js
index c03de9b..7f6c74b 100644
--- a/recipes/producthunt/webview.js
+++ b/recipes/producthunt/webview.js
@@ -1,7 +1,13 @@
-module.exports = (Ferdi) => {
+module.exports = Ferdi => {
+  const notificationsSelector = document.querySelector(
+    '[class*=header_] [class*=content_] [class*=actions_] [class*=notificationsButton_]',
+  );
+
   const getMessages = () => {
-    Ferdi.setBadge(document.querySelector('[class*=header_] [class*=content_] [class*=actions_] [class*=notificationsButton_]').innerText)
-  }
+    if (notificationsSelector) {
+      Ferdi.setBadge(notificationsSelector.textContent);
+    }
+  };
 
-  Ferdi.loop(getMessages)
-}
+  Ferdi.loop(getMessages);
+};