some small tweaks to the test: