fix failing tests.