fix broken tests.