Fixes the tests for MessageInputController