         if let userInfo = notification.userInfo {
             let animationDuration: NSTimeInterval = (userInfo[UIKeyboardAnimationDurationUserInfoKey] as! NSNumber).doubleValue
 
+            self.setConversaitonCollectionViewOriginalContentInset()
+
             UIView.animateWithDuration(animationDuration, delay: 0, options: .BeginFromCurrentState, animations: { () -> Void in
                 self.messageToolbarBottomConstraint.constant = 0
                 self.view.layoutIfNeeded()
-                }, completion: { (finished) -> Void in
+
+            }, completion: { (finished) -> Void in
             })
         }
     }

        conversationCollectionView.registerNib(UINib(nibName: chatLeftTextCellIdentifier, bundle: nil), forCellWithReuseIdentifier: chatLeftTextCellIdentifier)
        conversationCollectionView.registerNib(UINib(nibName: chatRightTextCellIdentifier, bundle: nil), forCellWithReuseIdentifier: chatRightTextCellIdentifier)

        var contentInset = conversationCollectionView.contentInset
        contentInset.bottom = messageToolbar.intrinsicContentSize().height
        conversationCollectionView.contentInset = contentInset

        messageToolbarBottomConstraint.constant = 0

        updateMessageToolbarWithKeyboardChange = true
    }


    // MARK: Keyboard



        return CGSizeMake(collectionViewWidth, 60)
    }

    func collectionView(collectionView: UICollectionView, didSelectItemAtIndexPath indexPath: NSIndexPath) {
        view.endEditing(true)
    }
}