fixing tests.