Fix a bug in the test.