Add missing yield statement