Made small change to yield