Generator: Fix missing key when yield