Fix bug in test.