Unittest tweaks