fix yield return demo