1 files changed, 12 insertions, 2 deletions

diff --git a/uncompressed/github/webview.js b/uncompressed/github/webview.js
index 0543d1f..c3e8729 100644
--- a/uncompressed/github/webview.js
+++ b/uncompressed/github/webview.js
@@ -2,9 +2,19 @@
 
 module.exports = Franz => {
   const getMessages = function getMessages() {
-    Franz.setBadge(document.querySelector('[class*="mail-status unread"]') !== null ? 1 : 0);
+    const directCountElement = document.querySelector('.filter-list .count');
+    const indirectCountElement = document.querySelector('[class*="mail-status unread"]')
+    let directCount, indirectCount
+    if (directCountElement) {
+      directCount = parseInt(directCountElement.innerHTML, 10);
+    } else {
+      if (indirectCountElement) {
+        indirectCount = 1;
+      }
+    }
+    Franz.setBadge(directCount, indirectCount);
   };
 
-  Franz.loop(0, getMessages);
+  Franz.loop(getMessages);
 }
 //# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IiIsImZpbGUiOiJnaXRodWIvd2Vidmlldy5qcyIsInNvdXJjZXNDb250ZW50IjpbXX0=
\ No newline at end of file