diff --git a/app/src/main/java/io/legado/app/ui/book/changesource/ChangeChapterSourceViewModel.kt b/app/src/main/java/io/legado/app/ui/book/changesource/ChangeChapterSourceViewModel.kt
index c11326d6e..5bf669a26 100644
--- a/app/src/main/java/io/legado/app/ui/book/changesource/ChangeChapterSourceViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/book/changesource/ChangeChapterSourceViewModel.kt
@@ -285,7 +285,9 @@ class ChangeChapterSourceViewModel(application: Application) : BaseViewModel(app
                     if (book.tocUrl.isEmpty()) {
                         WebBook.getBookInfoAwait(this, source, book)
                     }
-                    WebBook.getChapterListAwait(this, source, book)
+                    val toc = WebBook.getChapterListAwait(this, source, book)
+                    tocMap[book.bookUrl] = toc
+                    toc
                 }
         }.onSuccess {
             success(it)