Generator: Fixed return statement