fixed one of the tests