Fix bug for Arr::flatten