Fix doc string
Fix doc stirng