Modify the pb agent for gtest