Added missing yield