unit test tweak