Swift UITableView AccessoryType (галочка)

Question

Я использую аксессуар tableview.Type (галочка) в Tableview (didSelectRowAt :). Когда пользователь щелкает строку без галочки, галочка появится. Если щелкнуть строку с галочкой, галочка исчезнет. Ниже приведен код:

if cell?.accessoryType == .checkmark {
    cell?.accessoryType = .none
} else {
    cell?.accessoryType = .checkmark
}

Приведенный выше код работает должным образом. При изменении на следующее, это не работает. Есть идеи?

if cell?.accessoryType == .none {
   cell?.accessoryType = .checkmark
} else {
    cell?.accessoryType = .none
}

Код выглядит следующим образом:

func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
    let cell = tableView.cellForRow(at: indexPath)
    if cell?.accessoryType == .checkmark {
        cell?.accessoryType = .none
    } else {
        cell?.accessoryType = .checkmark
    }

}