1 files changed, 6 insertions, 1 deletions
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt b/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt
index 161ed93d..a37c2ee9 100644
--- a/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt
+++ b/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt
@@ -10,9 +10,14 @@ import org.koin.standalone.StandAloneContext
 
 interface FrostPrivateDao {
     fun cookieDao(): CookieDao
+    fun notifDao(): NotificationDao
 }
 
-@Database(entities = [CookieEntity::class], version = 1, exportSchema = true)
+@Database(
+    entities = [CookieEntity::class, NotificationInfoEntity::class, NotificationEntity::class],
+    version = 1,
+    exportSchema = true
+)
 abstract class FrostPrivateDatabase : RoomDatabase(), FrostPrivateDao {
     companion object {
         const val DATABASE_NAME = "frost-priv-db"