Could you just add this line after the `yield` statement, before closing the original block and without having to tap into a new one?