chat UI bug fixes