Allow yield as identifier in certain contexts