Fix doc comments