diff --git a/JSMessagesTableViewController/JSMessageInputView.h b/JSMessagesTableViewController/JSMessageInputView.h
index 1acb4bd..fadebf3 100644
--- a/JSMessagesTableViewController/JSMessageInputView.h
+++ b/JSMessagesTableViewController/JSMessageInputView.h
@@ -40,6 +40,9 @@
 @property (strong, nonatomic) UITextView *textView;
 @property (strong, nonatomic) UIButton *sendButton;
 
+#pragma mark - Initialization
+- (void)setupSendButton;
+
 #pragma mark - Message input view
 - (void)adjustTextViewHeightBy:(CGFloat)changeInHeight;
 + (CGFloat)textViewLineHeight;
diff --git a/JSMessagesTableViewController/JSMessageInputView.m b/JSMessagesTableViewController/JSMessageInputView.m
index 15e541e..8b23ffb 100644
--- a/JSMessagesTableViewController/JSMessageInputView.m
+++ b/JSMessagesTableViewController/JSMessageInputView.m
@@ -41,7 +41,6 @@
 
 - (void)setup;
 - (void)setupTextView;
-- (void)setupSendButton;
 
 @end