Fix doc test