Refine the unit test.