Quick fix for not updating Chat UI