Refreshed Polymock class with extra attributes