Fix yield (missing part)