Fix chat tests