Fixes regression in tests.