fixed one of the tests.