tweak unit testing