diff --git a/TagTunes/MainViewController.swift b/TagTunes/MainViewController.swift index a7a865c..3d98d94 100644 --- a/TagTunes/MainViewController.swift +++ b/TagTunes/MainViewController.swift @@ -684,20 +684,9 @@ extension MainViewController: NSUserInterfaceValidations { } private func canSave() -> Bool { - for row in outlineView.selectedRowIndexes where sectionOfRow(row) == .Albums { - let item = outlineView.itemAtRow(row) - if let album = item as? Album { - if !album.saved { - return true - } - } else if let track = item as? Track { - if !track.saved { - return true - } - } else if let track = item as? iTunesTrack { - if parentForTrack(track)?.saved == false { - return true - } + for row in outlineView.selectedRowIndexes { + if sectionOfRow(row) == .Albums { + return true } } return false