Fix code block in return_with_argument_inside_generator.