Fixed the tests.