a minor fix in the test code