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

    }
    
    public func didEndDisplaying(saveButton: SaveButton, for article: WMFArticle) {
        guard let key = article.key else {
            return
        }
        let tag = key.hash

    fileprivate var updatedArticle: WMFArticle?
    
    @objc func articleUpdated(notification: Notification) {
        guard let article = notification.object as? WMFArticle, let key = article.inMemoryKey, let saveButtons = visibleSaveButtons[key.hash] else {
            return
        }
        for saveButton in saveButtons {
