diff --git a/Source/AsyncDisplayKit+IGListKitMethods.m b/Source/AsyncDisplayKit+IGListKitMethods.m index 7e61c2806..f95d65287 100644 --- a/Source/AsyncDisplayKit+IGListKitMethods.m +++ b/Source/AsyncDisplayKit+IGListKitMethods.m @@ -13,6 +13,8 @@ #import "AsyncDisplayKit+IGListKitMethods.h" #import #import +#import + @implementation ASIGListSectionControllerMethods @@ -35,7 +37,7 @@ + (__kindof UICollectionReusableView *)viewForSupplementaryElementOfKind:(NSStri atIndex:(NSInteger)index sectionController:(IGListSectionController *)sectionController { - return [sectionController.collectionContext dequeueReusableSupplementaryViewOfKind:elementKind forSectionController:sectionController class:[UICollectionReusableView class] atIndex:index]; + return [sectionController.collectionContext dequeueReusableSupplementaryViewOfKind:elementKind forSectionController:sectionController class:[_ASCollectionReusableView class] atIndex:index]; } + (CGSize)sizeForSupplementaryViewOfKind:(NSString *)elementKind atIndex:(NSInteger)index