diff --git a/Source/IGListKit/Internal/IGListAdapterUpdaterHelpers.m b/Source/IGListKit/Internal/IGListAdapterUpdaterHelpers.m
index 5d0123ae..fcc8dee1 100644
--- a/Source/IGListKit/Internal/IGListAdapterUpdaterHelpers.m
+++ b/Source/IGListKit/Internal/IGListAdapterUpdaterHelpers.m
@@ -7,8 +7,7 @@
 
 #import "IGListAdapterUpdaterHelpers.h"
 
-#import <IGAssert/IGAssert.h>
-
+#import <IGListDiffKit/IGListAssert.h>
 #import <IGListDiffKit/IGListBatchUpdateData.h>
 #import <IGListDiffKit/IGListIndexSetResult.h>
 #import <IGListDiffKit/IGListDiffable.h>
diff --git a/Tests/Objects/IGListTestUICollectionViewDataSource.m b/Tests/Objects/IGListTestUICollectionViewDataSource.m
index 2dfbfbad..d6be4732 100644
--- a/Tests/Objects/IGListTestUICollectionViewDataSource.m
+++ b/Tests/Objects/IGListTestUICollectionViewDataSource.m
@@ -9,7 +9,7 @@
 
 #import "IGListTestUICollectionViewDataSource.h"
 
-#import <IGAssert/IGAssert.h>
+#import <IGListDiffKit/IGListAssert.h>
 
 @implementation IGSectionObject {
     NSString *_identifier;