Update unittest case to test new design.