diff --git a/app/src/main/java/io/legado/app/ui/book/explore/ExploreShowViewModel.kt b/app/src/main/java/io/legado/app/ui/book/explore/ExploreShowViewModel.kt
index 4bcc43855..59ca1783a 100644
--- a/app/src/main/java/io/legado/app/ui/book/explore/ExploreShowViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/book/explore/ExploreShowViewModel.kt
@@ -87,9 +87,8 @@ class ExploreShowViewModel(application: Application) : BaseViewModel(application
         val name = book.name
         val author = book.author
         val bookUrl = book.bookUrl
-        return (author.isNotBlank() && bookshelf.contains("$name-$author"))
-                || bookshelf.contains(name)
-                || bookshelf.contains(bookUrl)
+        val key = if (author.isNotBlank()) "$name-$author" else name
+        return bookshelf.contains(key) || bookshelf.contains(bookUrl)
     }
 
 }
diff --git a/app/src/main/java/io/legado/app/ui/book/search/SearchViewModel.kt b/app/src/main/java/io/legado/app/ui/book/search/SearchViewModel.kt
index 426186af8..53b91af12 100644
--- a/app/src/main/java/io/legado/app/ui/book/search/SearchViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/book/search/SearchViewModel.kt
@@ -88,9 +88,8 @@ class SearchViewModel(application: Application) : BaseViewModel(application) {
         val name = book.name
         val author = book.author
         val bookUrl = book.bookUrl
-        return (author.isNotBlank() && bookshelf.contains("$name-$author"))
-                || bookshelf.contains(name)
-                || bookshelf.contains(bookUrl)
+        val key = if (author.isNotBlank()) "$name-$author" else name
+        return bookshelf.contains(key) || bookshelf.contains(bookUrl)
     }
 
     /**