fixed the unit test