From bdf73da0ba9f1565a5a858d25f7dd31b2b0f2a47 Mon Sep 17 00:00:00 2001
From: Jay <jays2kings@gmail.com>
Date: Mon, 13 Apr 2020 22:42:41 -0400
Subject: [PATCH] Fixed unread badge type issues

---
 .../eu/kanade/tachiyomi/ui/library/LibraryPresenter.kt     | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/eu/kanade/tachiyomi/ui/library/LibraryPresenter.kt b/app/src/main/java/eu/kanade/tachiyomi/ui/library/LibraryPresenter.kt
index e52e87d720..ded7f9b5ee 100644
--- a/app/src/main/java/eu/kanade/tachiyomi/ui/library/LibraryPresenter.kt
+++ b/app/src/main/java/eu/kanade/tachiyomi/ui/library/LibraryPresenter.kt
@@ -87,7 +87,10 @@ class LibraryPresenter(
             totalChapters = null
             val mangaMap = withContext(Dispatchers.IO) {
                 val library = getLibraryFromDB()
-                library.apply { setDownloadCount(library) }
+                library.apply {
+                    setDownloadCount(library)
+                    setUnreadBadge(library)
+                }
                 allLibraryItems = library
                 var mangaMap = library
                 mangaMap = applyFilters(mangaMap)
@@ -477,7 +480,7 @@ class LibraryPresenter(
         launchUI {
             val mangaMap = allLibraryItems
             withContext(Dispatchers.IO) { setUnreadBadge(mangaMap) }
-            libraryItems = mangaMap
+            allLibraryItems = mangaMap
             val current = libraryItems
             withContext(Dispatchers.IO) { setUnreadBadge(current) }
             libraryItems = current