replace yield_context_2 with yield_context