we need the `flatten`?