doc test fix