Small fix in tests.