Small fix for tests.