     }
     
     func configureWithMessage(message: Message, mediaTapAction: MediaTapAction?) {
+
+        self.message = message
 
         self.mediaTapAction = mediaTapAction
 


        avatarImageViewWidthConstraint.constant = YepConfig.chatCellAvatarSize()

        gapBetweenDotImageViewAndBubbleConstraint.constant = YepConfig.ChatCell.gapBetweenDotImageViewAndBubble
        
        mapImageView.tintColor = UIColor.rightBubbleTintColor()

        mapImageView.userInteractionEnabled = true

import UIKit
import MapKit

class ChatRightLocationCell: ChatRightBaseCell {

    @IBOutlet weak var avatarImageView: UIImageView!
    @IBOutlet weak var avatarImageViewWidthConstraint: NSLayoutConstraint!

    @IBOutlet weak var mapImageView: UIImageView!

    typealias MediaTapAction = () -> Void
    var mediaTapAction: MediaTapAction?

