Fix yield expression parsing.