Improve a unit tests.