Reload items along with other updates in batch update block

Author: crazytonyli

Pod/Classes/WPMediaPickerViewController.m

@@ -688,6 +688,9 @@ - (void)updateDataWithRemoved:(NSIndexSet *)removed inserted:(NSIndexSet *)inser
         if ([inserted count] > 0) {
             [self.collectionView insertItemsAtIndexPaths:[self indexPathsFromIndexSet:inserted section:0]];
         }
+        if ([changed count] > 0) {
+            [self.collectionView reloadItemsAtIndexPaths:[self indexPathsFromIndexSet:changed section:0]];
+        }
         for (id<WPMediaMove> move in moves) {
             [self.collectionView moveItemAtIndexPath:[NSIndexPath indexPathForItem:[move from] inSection:0]
                                          toIndexPath:[NSIndexPath indexPathForItem:[move to] inSection:0]];
@@ -697,9 +700,7 @@ - (void)updateDataWithRemoved:(NSIndexSet *)removed inserted:(NSIndexSet *)inser
         }
     } completion:^(BOOL finished) {
         [weakSelf refreshSelection];
-        NSMutableSet<NSIndexPath *> *indexPaths = [NSMutableSet setWithArray:[weakSelf indexPathsFromIndexSet:changed section:0]];
-        [indexPaths addObjectsFromArray:weakSelf.collectionView.indexPathsForSelectedItems];
-        [weakSelf.collectionView reloadItemsAtIndexPaths:[indexPaths allObjects]];
+        [weakSelf.collectionView reloadItemsAtIndexPaths:weakSelf.collectionView.indexPathsForSelectedItems];
     }];
 
 }