Better unit test setup