Minor change to the test code.