Refactor iterator inject method