             if let index = previewAttachmentPhotos.indexOf(previewAttachmentPhoto) {
                 return previewTransitionViews?[index]
             }
+
+        } else if photo is PreviewMessagePhoto {
+            return (previewTransitionViews?.first)!
         }
 
         return nil


                        let photosViewController = PhotosViewController(photos: photos, initialPhoto: initialPhoto, delegate: self)
                        self.presentViewController(photosViewController, animated: true, completion: nil)

                        //vc.previewMedias = mediaMessages.map({ PreviewMedia.MessageType(message: $0) })
                        //vc.startIndex = index
                    }
                }



                /*
                let vc = UIStoryboard(name: "MediaPreview", bundle: nil).instantiateViewControllerWithIdentifier("MediaPreviewViewController") as! MediaPreviewViewController


        previewTransitionViews = nil
        previewAttachmentPhotos = []
        previewMessagePhotos = []
    }
}

