Ammend example code

tmdvs · Feb 4, 2017 · 8bd1021 · 8bd1021
1 parent 2d90e0a
commit 8bd1021
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/TDBadgedCell/ViewController.swift b/TDBadgedCell/ViewController.swift
@@ -28,9 +28,9 @@ class ViewController: UITableViewController {
         return demoItems.count
     }
 
-    override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> TDBadgedCell {
-        var cell : TDBadgedCell? = tableView.dequeueReusableCell(withIdentifier:"BadgedCell") as! TDBadgedCell?;
-        if((cell == nil)) {
+    override func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
+        var cell = tableView.dequeueReusableCell(withIdentifier:"BadgedCell") as? TDBadgedCell;
+        if(cell == nil) {
             cell = TDBadgedCell(style: .default, reuseIdentifier: "BadgedCell");
         }