diff --git a/app/src/main/java/io/legado/app/data/AppDatabase.kt b/app/src/main/java/io/legado/app/data/AppDatabase.kt
index 15ab3e0c0..6b1c46817 100644
--- a/app/src/main/java/io/legado/app/data/AppDatabase.kt
+++ b/app/src/main/java/io/legado/app/data/AppDatabase.kt
@@ -10,14 +10,11 @@ import io.legado.app.data.dao.BookDao
 import io.legado.app.data.dao.BookGroupDao
 import io.legado.app.data.dao.BookSourceDao
 import io.legado.app.data.dao.ReplaceRuleDao
-import io.legado.app.data.entities.Book
-import io.legado.app.data.entities.BookSource
-import io.legado.app.data.entities.Chapter
-import io.legado.app.data.entities.ReplaceRule
+import io.legado.app.data.entities.*
 
 
 @Database(
-    entities = [Book::class, BookSource::class, Chapter::class, ReplaceRule::class],
+    entities = [Book::class, BookGroup::class, BookSource::class, Chapter::class, ReplaceRule::class],
     version = 1,
     exportSchema = true
 )