4 files changed, 12 insertions, 10 deletions

diff --git a/recipes/instagram-direct-messages/package.json b/recipes/instagram-direct-messages/package.json
index 41aa129..ab94fae 100644
--- a/recipes/instagram-direct-messages/package.json
+++ b/recipes/instagram-direct-messages/package.json
@@ -1,7 +1,7 @@
 {
   "id": "instagram-direct-messages",
   "name": "Instagram Direct Messages",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "license": "MIT",
   "config": {
     "serviceURL": "https://www.instagram.com/direct/inbox/",
diff --git a/recipes/instagram-direct-messages/webview.js b/recipes/instagram-direct-messages/webview.js
index 01187b8..eb33ed1 100644
--- a/recipes/instagram-direct-messages/webview.js
+++ b/recipes/instagram-direct-messages/webview.js
@@ -12,9 +12,11 @@ setInterval(() => {
 
 module.exports = Ferdium => {
   const getMessages = () => {
-    const element = document.querySelector('a[href^="/direct/inbox"]');
+    const element = document.querySelector('a[href^="/direct/inbox"] span');
     Ferdium.setBadge(
-      element.textContent ? Ferdium.safeParseInt(element.textContent) : 0,
+      element && element.textContent
+        ? Ferdium.safeParseInt(element.textContent)
+        : 0,
     );
   };
 
diff --git a/recipes/instagram/package.json b/recipes/instagram/package.json
index e0449bb..7425164 100644
--- a/recipes/instagram/package.json
+++ b/recipes/instagram/package.json
@@ -1,7 +1,7 @@
 {
   "id": "instagram",
   "name": "Instagram",
-  "version": "2.5.2",
+  "version": "2.5.3",
   "license": "MIT",
   "config": {
     "serviceURL": "https://instagram.com/",
diff --git a/recipes/instagram/webview.js b/recipes/instagram/webview.js
index b63e683..3092d21 100644
--- a/recipes/instagram/webview.js
+++ b/recipes/instagram/webview.js
@@ -40,12 +40,12 @@ module.exports = (Ferdium, settings) => {
   );
 
   const getMessages = () => {
-    const element = document.querySelector('a[href^="/direct/inbox"]');
-    if (element) {
-      Ferdium.setBadge(
-        element.textContent ? Ferdium.safeParseInt(element.textContent) : 0,
-      );
-    }
+    const element = document.querySelector('a[href^="/direct/inbox"] span');
+    Ferdium.setBadge(
+      element && element.textContent
+        ? Ferdium.safeParseInt(element.textContent)
+        : 0,
+    );
   };
 
   Ferdium.loop(getMessages);