Changed `flatten_once` into `flatten(x)`