Make unit test more compact