From b449e05edcea21ba6274c699eb2a597047f33f45 Mon Sep 17 00:00:00 2001
From: Max Sokolov <i@maxsokolov.net>
Date: Sat, 15 Oct 2016 12:06:48 +0300
Subject: [PATCH] fix

---
 Sources/TableDirector.swift | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Sources/TableDirector.swift b/Sources/TableDirector.swift
index 665e481..63de5f7 100644
--- a/Sources/TableDirector.swift
+++ b/Sources/TableDirector.swift
@@ -36,6 +36,8 @@ open class TableDirector: NSObject, UITableViewDataSource, UITableViewDelegate {
         didSet {
             if shouldUsePrototypeCellHeightCalculation {
                 rowHeightCalculator = TablePrototypeCellHeightCalculator(tableView: tableView)
+            } else {
+                rowHeightCalculator = nil
             }
         }
     }