Small fix to tests.