diff --git a/Source/ASSectionController.h b/Source/ASSectionController.h
index dbf39d93c0..f57413f07d 100644
--- a/Source/ASSectionController.h
+++ b/Source/ASSectionController.h
@@ -7,9 +7,13 @@
 //
 
 #import <Foundation/Foundation.h>
+#import <AsyncDisplayKit/ASBlockTypes.h>
+#import <AsyncDisplayKit/ASDimension.h>
 
 NS_ASSUME_NONNULL_BEGIN
 
+@class ASBatchContext;
+
 /**
  * A protocol that your section controllers should conform to,
  * in addition to IGListSectionType, in order to be used with AsyncDisplayKit.
diff --git a/Source/ASSupplementaryNodeSource.h b/Source/ASSupplementaryNodeSource.h
index 9d3233c9e2..52ad429377 100644
--- a/Source/ASSupplementaryNodeSource.h
+++ b/Source/ASSupplementaryNodeSource.h
@@ -7,12 +7,11 @@
 //
 
 #import <Foundation/Foundation.h>
+#import <AsyncDisplayKit/ASBlockTypes.h>
 #import <AsyncDisplayKit/ASDimension.h>
 
 NS_ASSUME_NONNULL_BEGIN
 
-@class ASCellNode;
-
 @protocol ASSupplementaryNodeSource <NSObject>
 
 /**