improve the unit test