update the unit test