improving a unit test