Small fix to test.