Fix yield -> reutnr