Fixing protobuf after refactory