diff --git a/RATreeView/RATreeView/RATreeView.h b/RATreeView/RATreeView/RATreeView.h
index 5b95eaf..0ff7cb9 100644
--- a/RATreeView/RATreeView/RATreeView.h
+++ b/RATreeView/RATreeView/RATreeView.h
@@ -546,7 +546,7 @@ typedef enum RATreeViewRowAnimation {
 /// @name Accessing Cells
 ///------------------------------------------------
 
-- (nullable UITableViewCell *)cellForItem:(id)item;
+- (nullable __kindof UITableViewCell *)cellForItem:(id)item;
 - (nullable NSArray *)visibleCells;
 - (nullable id)itemForCell:(UITableViewCell *)cell;
 - (nullable id)itemForRowAtPoint:(CGPoint)point;
diff --git a/RATreeView/RATreeView/RATreeView.m b/RATreeView/RATreeView/RATreeView.m
index 58b2497..e7f2c7c 100644
--- a/RATreeView/RATreeView/RATreeView.m
+++ b/RATreeView/RATreeView/RATreeView.m
@@ -404,7 +404,7 @@ - (id)parentForItem:(id)item
 
 #pragma mark - Accessing Cells
 
-- (UITableViewCell *)cellForItem:(id)item
+- (__kindof UITableViewCell *)cellForItem:(id)item
 {
   NSIndexPath *indexPath = [self indexPathForItem:item];
   return [self.tableView cellForRowAtIndexPath:indexPath];