Use existing yield_expr definition