         guard let article = updatedArticle else {
             return
         }
-        guard activeKey == article.key else {
+        guard activeKey == article.inMemoryKey else {
             return
         }
         let tag = activeKey?.hash ?? 0

class SaveButtonsController: NSObject, SaveButtonDelegate {
    
    var visibleSaveButtons = [Int: Set<SaveButton>]()
    var visibleArticleKeys = [Int: String]()
    var visibleUserInfo = [Int: Any]()
    
    let dataStore: MWKDataStore
    let savedPagesFunnel = SavedPagesFunnel()
    var activeSender: SaveButton?
    var activeKey: String?
    
    required init(dataStore: MWKDataStore) {
        self.dataStore = dataStore

    }
    
    public func willDisplay(saveButton: SaveButton, for article: WMFArticle, with userInfo: Any? = nil) {
        guard let key = article.inMemoryKey else {
            return
        }
        let tag = key.hash
