added back the missing classes in the ProtoBase