improve a unit test