         
         PhotoManager.sharedInstance.assetToImage(asset: asset, imageSize: size, isCliping: false) { image in
             self.currentImage = image
-            self.dismiss(animated: true, completion: nil)
+            self.dismiss(animated: animated, completion: {
+                if !animated {
+                    parentVC?.delegate?.getAssets(assets: PhotoManager.sharedInstance.assets)
+                    parentVC?.delegate?.dismiss()
+                }
+            })
         }
     }
 }

//  IBAction
private extension CSPhotoGalleryDetailViewController {
    @IBAction func backBtnAction(_ sender: Any) {
        dismiss()
    }
    
    @IBAction func checkBtnAction(_ sender: Any) {


//  MARK:- Extension
fileprivate extension CSPhotoGalleryDetailViewController {
    func dismiss(animated: Bool) {
        let parentVC = presentingViewController as? CSPhotoGalleryViewController
        parentVC?.scrollRectToVisible(indexPath: currentIndexPath)
        
