fix test after 'with' statement change