Small fix for test.