Fix issue found in test.