Fixed doc string so that it works on subsequent runs