Add missing yield