Minor change to doc string