Fixing UI of the message input