Small fix for test suite.