Remove extra '.first.flatten'