adjust  the unit test