Fix bug with yield as param in last stmt of method