diff --git a/Source/ImagePickerController.swift b/Source/ImagePickerController.swift index 7953fe4a..3adcfe0f 100644 --- a/Source/ImagePickerController.swift +++ b/Source/ImagePickerController.swift @@ -158,7 +158,7 @@ public class ImagePickerController: UIViewController { object: nil) NSNotificationCenter.defaultCenter().addObserver(self, - selector: "adjustButtonTitle:", + selector: "didReloadAssets:", name: ImageStack.Notifications.stackDidReload, object: nil) @@ -168,6 +168,12 @@ public class ImagePickerController: UIViewController { object: nil) } + func didReloadAssets(notification: NSNotification) { + adjustButtonTitle(notification) + galleryView.collectionView.reloadData() + galleryView.collectionView.setContentOffset(CGPointZero, animated: false) + } + func volumeChanged(notification: NSNotification) { guard let slider = volumeView.subviews.filter({ $0 is UISlider }).first as? UISlider, let userInfo = notification.userInfo,