Changing unit test setup