Fix display for new chat layout