Better unit test