Adjust for protobuf change