Unittest fix/tweak