diff --git a/AsyncDisplayKit/Private/NSArray+Diffing.m b/AsyncDisplayKit/Private/NSArray+Diffing.m
index 00893d1416..d7a9fd3a52 100644
--- a/AsyncDisplayKit/Private/NSArray+Diffing.m
+++ b/AsyncDisplayKit/Private/NSArray+Diffing.m
@@ -53,7 +53,7 @@
     for (NSInteger j = array.count; j >= 0; j--) {
       if (i == self.count || j == array.count) {
         lengths[i][j] = 0;
-      } else if ([self[i] isEqual:array[j]]) {
+      } else if (comparison(self[i], array[j])) {
         lengths[i][j] = 1 + lengths[i+1][j+1];
       } else {
         lengths[i][j] = MAX(lengths[i+1][j], lengths[i][j+1]);