fixing a unit test