fixing the unit test