Small improvement to the tests.