Fix doc comment