Modification in doc test