tweak unit test a bit