diff --git a/Sources/MessagesCollectionView.swift b/Sources/MessagesCollectionView.swift
index b8cff960..8d556ca7 100644
--- a/Sources/MessagesCollectionView.swift
+++ b/Sources/MessagesCollectionView.swift
@@ -49,6 +49,9 @@ open class MessagesCollectionView: UICollectionView {
     override public init(frame: CGRect, collectionViewLayout layout: UICollectionViewLayout) {
         super.init(frame: frame, collectionViewLayout: layout)
         backgroundColor = .white
+        // https://stackoverflow.com/questions/31049651/uitextview-as-inputaccessoryview-doesnt-render-text-until-after-animation
+        // Calling this here avoids a side effect where the inputAccessoryView dissapears after tap
+        snapshotView(afterScreenUpdates: true)
     }
 
     required public init?(coder aDecoder: NSCoder) {
diff --git a/Sources/MessagesViewController.swift b/Sources/MessagesViewController.swift
index 822ea0f4..3bf01745 100644
--- a/Sources/MessagesViewController.swift
+++ b/Sources/MessagesViewController.swift
@@ -58,8 +58,6 @@ open class MessagesViewController: UIViewController {
 		registerReusableViews()
 		setupDelegates()
 
-        // https://stackoverflow.com/questions/31049651/uitextview-as-inputaccessoryview-doesnt-render-text-until-after-animation
-        inputAccessoryView?.snapshotView(afterScreenUpdates: true)
 	}
 
     override open func viewDidLayoutSubviews() {