diff --git a/Sources/TableDirector.swift b/Sources/TableDirector.swift index c17ce2c..1a7724c 100644 --- a/Sources/TableDirector.swift +++ b/Sources/TableDirector.swift @@ -60,7 +60,7 @@ open class TableDirector: NSObject, UITableViewDataSource, UITableViewDelegate { self.cellRegisterer = TableCellRegisterer(tableView: tableView) } - self.rowHeightCalculator = cellHeightCalculator + self.rowHeightCalculator = cellHeightCalculator ?? ExpandableCellHeightCalculator(tableView: tableView) self.scrollDelegate = scrollDelegate self.tableView = tableView self.tableView?.delegate = self