tweak unit test