diff --git a/app/src/main/java/io/legado/app/ui/book/toc/TocScreen.kt b/app/src/main/java/io/legado/app/ui/book/toc/TocScreen.kt
index 2220b6c51..3d7e08d1b 100644
--- a/app/src/main/java/io/legado/app/ui/book/toc/TocScreen.kt
+++ b/app/src/main/java/io/legado/app/ui/book/toc/TocScreen.kt
@@ -152,8 +152,8 @@ fun TocScreen(
 
     var editingBookmark by remember { mutableStateOf<Bookmark?>(null) }
 
-    val useReplace = viewModel.useReplace
-    val showWordCount = viewModel.showWordCount
+    val useReplace = state.useReplace
+    val showWordCount = state.showWordCount
     val bookmarkManagementTitle = stringResource(R.string.bookmark_management)
     val locateCurrentReadingText = stringResource(R.string.locate_current_reading)
     val moveToTopText = stringResource(R.string.move_to_top)
@@ -686,7 +686,7 @@ fun ChapterListContent(
                             .animateItem()
                             .fillMaxWidth(),
                         item = uiItem,
-                        showWordCount = viewModel.showWordCount,
+                        showWordCount = state.showWordCount,
                         onClick = {
                             if (state.selectedIds.isNotEmpty())
                                 viewModel.toggleSelection(uiItem.id)
diff --git a/app/src/main/java/io/legado/app/ui/book/toc/TocViewModel.kt b/app/src/main/java/io/legado/app/ui/book/toc/TocViewModel.kt
index 6507c3073..40249f30e 100644
--- a/app/src/main/java/io/legado/app/ui/book/toc/TocViewModel.kt
+++ b/app/src/main/java/io/legado/app/ui/book/toc/TocViewModel.kt
@@ -80,7 +80,9 @@ data class TocActionState(
     override val searchKey: String = "",
     override val isSearch: Boolean = false,
     override val isLoading: Boolean = false,
-    val downloadSummary: String = ""
+    val downloadSummary: String = "",
+    val useReplace: Boolean = false,
+    val showWordCount: Boolean = true,
 ) : ListUiState<TocItemUi>
 
 data class TocDomainItem(
@@ -347,7 +349,9 @@ class TocViewModel(
             searchKey = _searchKey.value,
             isSearch = isSearch,
             isLoading = isUploading,
-            downloadSummary = downloadSummary.value
+            downloadSummary = downloadSummary.value,
+            useReplace = tocPreferences.value.useReplace,
+            showWordCount = tocPreferences.value.showWordCount,
         )
     }