     }
     
     private func registerCell(with identifier: String) {
-        if let style = modelDelegate?.cellStyle(forIdentifier: identifier) {
-            switch style {
-            case .default:
-                tableView.register(DefaultCell.self, forCellReuseIdentifier: identifier)
-            case .subtitle:
-                tableView.register(SubtitleCell.self, forCellReuseIdentifier: identifier)
-            case .leftDetail:
-                tableView.register(LeftDetailCell.self, forCellReuseIdentifier: identifier)
-            case .rightDetail:
-                tableView.register(RightDetailCell.self, forCellReuseIdentifier: identifier)
-            case .toggle:
-                tableView.register(ToggleCell.self, forCellReuseIdentifier: identifier)
-            case .customClass(let cellClass):
-                tableView.register(cellClass, forCellReuseIdentifier: identifier)
-            case .customNib(let cellNib):
-                tableView.register(cellNib, forCellReuseIdentifier: identifier)
-            }
-        } else {
+        switch cellStyle(forIdentifier: identifier) {
+        case .default:
             tableView.register(DefaultCell.self, forCellReuseIdentifier: identifier)
+        case .subtitle:
+            tableView.register(SubtitleCell.self, forCellReuseIdentifier: identifier)
+        case .leftDetail:
+            tableView.register(LeftDetailCell.self, forCellReuseIdentifier: identifier)
+        case .rightDetail:
+            tableView.register(RightDetailCell.self, forCellReuseIdentifier: identifier)
+        case .toggle:
+            tableView.register(ToggleCell.self, forCellReuseIdentifier: identifier)
+        case .customClass(let cellClass):
+            tableView.register(cellClass, forCellReuseIdentifier: identifier)
+        case .customNib(let cellNib):
+            tableView.register(cellNib, forCellReuseIdentifier: identifier)
         }
     }
     

        (cell as? ToggleCell)?.delegate = self
        
        if let cell = cell as? TableModelCell {
            modelDelegate?.configureCell(cell, with: item)
        }
        
        return cell

import UIKit

open class TableModelViewController: UITableViewController {
    public var model: TableModel?
    
    required public init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)
