Minor tweaks to unit tests

just some very minor tweaks to unit tests, after #848 got merged in