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];