diff --git a/Example/Sources/View Controllers/AdvancedExampleViewController.swift b/Example/Sources/View Controllers/AdvancedExampleViewController.swift
index adeaae93..4892390f 100644
--- a/Example/Sources/View Controllers/AdvancedExampleViewController.swift	
+++ b/Example/Sources/View Controllers/AdvancedExampleViewController.swift	
@@ -171,6 +171,7 @@ final class AdvancedExampleViewController: ChatViewController {
     messageInputBar.inputTextView.layer.masksToBounds = true
     messageInputBar.inputTextView.scrollIndicatorInsets = UIEdgeInsets(top: 8, left: 0, bottom: 8, right: 0)
     configureInputBarItems()
+    inputBarType = .custom(messageInputBar)
   }
 
   // MARK: - Helpers