2 files changed, 5 insertions, 3 deletions
diff --git a/Sora/Data/Settings/Bookmark.swift b/Sora/Data/Settings/Bookmark.swift
index 2e6fc93..16f1e50 100644
--- a/Sora/Data/Settings/Bookmark.swift
+++ b/Sora/Data/Settings/Bookmark.swift
@@ -4,10 +4,12 @@ struct Bookmark: Codable, Identifiable, Hashable {
     let id: UUID
     let tags: [String]
     let createdAt: Date
+    let provider: BooruProvider
 
-    init(id: UUID = UUID(), tags: [String]) {
+    init(id: UUID = UUID(), provider: BooruProvider, tags: [String]) {
         createdAt = Date()
         self.id = id
         self.tags = tags
+        self.provider = provider
     }
 }
diff --git a/Sora/Data/Settings/Settings.swift b/Sora/Data/Settings/Settings.swift
index e8d1510..3ffa114 100644
--- a/Sora/Data/Settings/Settings.swift
+++ b/Sora/Data/Settings/Settings.swift
@@ -43,10 +43,10 @@ class Settings: ObservableObject {
         showNSFWPosts = false
     }
 
-    func addBookmark(tags: [String]) {
+    func addBookmark(provider: BooruProvider, tags: [String]) {
         var currentBookmarks = bookmarks
 
-        currentBookmarks.append(Bookmark(tags: tags.map { $0.lowercased() }))
+        currentBookmarks.append(Bookmark(provider: provider, tags: tags.map { $0.lowercased() }))
 
         bookmarks = currentBookmarks
     }