Fixed bug in yield.