Fix regression in tests.