tweaks to the unit tests