A little tweaking of the test code.