Correct the bug introduced by the new tests.

 This has to do with the fact ast.Yield is an Expr.