Improve a unit test