Refactoring

nek023 · Apr 6, 2015 · 274c5a5 · 274c5a5
1 parent e49b4d6
commit 274c5a5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/QBImagePicker/QBAssetsViewController.m b/QBImagePicker/QBAssetsViewController.m
@@ -112,8 +112,8 @@ - (void)setAssetsGroup:(ALAssetsGroup *)assetsGroup
 
     [self updateAssets];
 
-    // Find previous selected item if needed
-    if ([self isAutoDeselectEnabled]) {
+    if ([self isAutoDeselectEnabled] && self.imagePickerController.selectedAssetURLs.count > 0) {
+        // Get index of previous selected asset
         NSURL *previousSelectedAssetURL = [self.imagePickerController.selectedAssetURLs firstObject];
 
         [self.assets enumerateObjectsUsingBlock:^(ALAsset *asset, NSUInteger index, BOOL *stop) {
@@ -128,7 +128,7 @@ - (void)setAssetsGroup:(ALAssetsGroup *)assetsGroup
 
     [self.collectionView reloadData];
 }
-//
+
 - (BOOL)isAutoDeselectEnabled
 {
     return (self.imagePickerController.maximumNumberOfSelection == 1