From 1456406e7afa99f6e0be2c3c691f76ebbdaa1b16 Mon Sep 17 00:00:00 2001
From: pera <pera@xmartlabs.com>
Date: Tue, 23 May 2017 09:47:25 -0300
Subject: [PATCH] Fixes #977 (#14)

---
 Sources/ImageCell.swift | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/Sources/ImageCell.swift b/Sources/ImageCell.swift
index 68a7f4e..f9df2e2 100644
--- a/Sources/ImageCell.swift
+++ b/Sources/ImageCell.swift
@@ -30,10 +30,12 @@ public final class ImageCell: Cell<UIImage>, CellType {
         super.setup()
         
         accessoryType = .none
+        editingAccessoryView = .none
         let imageView = UIImageView(frame: CGRect(x: 0, y: 0, width: 44, height: 44))
         imageView.contentMode = .scaleAspectFill
         imageView.clipsToBounds = true
         accessoryView = imageView
+        editingAccessoryView = imageView
     }
     
     public override func update() {
@@ -41,6 +43,7 @@ public final class ImageCell: Cell<UIImage>, CellType {
         
         selectionStyle = row.isDisabled ? .none : .default
         (accessoryView as? UIImageView)?.image = row.value
+        (editingAccessoryView as? UIImageView)?.image = row.value
     }
 
 }