Fix chat ui