Adjust unittest