diff --git a/AsyncDisplayKit/Details/ASCollectionDataController.mm b/AsyncDisplayKit/Details/ASCollectionDataController.mm index 4de5bd3ca6..03b981a2a3 100644 --- a/AsyncDisplayKit/Details/ASCollectionDataController.mm +++ b/AsyncDisplayKit/Details/ASCollectionDataController.mm @@ -80,7 +80,7 @@ - (ASDisplayNode *)supplementaryNodeOfKind:(NSString *)kind atIndexPath:(NSIndexPath *)indexPath { ASDisplayNodeAssertMainThread(); - return [self internalCompletedNodes][kind][indexPath.section][indexPath.item]; + return [self completedNodesOfKind:kind][indexPath.section][indexPath.item]; } - (id)collectionDataSource diff --git a/AsyncDisplayKit/Details/ASDataController+Subclasses.h b/AsyncDisplayKit/Details/ASDataController+Subclasses.h index 84f07bf34d..fda6e69cca 100644 --- a/AsyncDisplayKit/Details/ASDataController+Subclasses.h +++ b/AsyncDisplayKit/Details/ASDataController+Subclasses.h @@ -35,7 +35,7 @@ */ - (void)willReloadData; -- (NSMutableDictionary *)internalCompletedNodes; +- (NSArray *)completedNodesOfKind:(NSString *)kind; /** * Measure and layout the given nodes in optimized batches, constraining each to a given size. diff --git a/AsyncDisplayKit/Details/ASDataController.mm b/AsyncDisplayKit/Details/ASDataController.mm index b55b62e528..94592578c8 100644 --- a/AsyncDisplayKit/Details/ASDataController.mm +++ b/AsyncDisplayKit/Details/ASDataController.mm @@ -781,9 +781,9 @@ static void *kASSizingQueueContext = &kASSizingQueueContext; #pragma mark - Data Querying (Subclass API) -- (NSMutableDictionary *)internalCompletedNodes +- (NSArray *)completedNodesOfKind:(NSString *)kind { - return _completedNodes; + return _completedNodes[kind]; } #pragma mark - Data Querying (External API)