Corrections in doc test