Fix doc string