Fix comment for doc