add UTs for PB types
add UTs for PB types