Remove unnecessary 'flatten'