minor tweaks in tests

minor tweak to unit tests after #848 was merged