re-add yield statement