diff --git a/TableKit/TKTableModel.m b/TableKit/TKTableModel.m index c0d020d..91c9790 100644 --- a/TableKit/TKTableModel.m +++ b/TableKit/TKTableModel.m @@ -108,7 +108,7 @@ - (void)didSelectRowAtIndexPath:(NSIndexPath *)indexPath { TKCellMapping *cellMapping = [self cellMappingForObject:object]; if (nil != cellMapping.onSelectRowBlock) { - UITableViewCell *cell = [self cellForRowAtIndexPath:indexPath]; + UITableViewCell *cell = [self.tableView cellForRowAtIndexPath:indexPath]; cellMapping.onSelectRowBlock(cell, object, indexPath); } }