Fixing one of the tests.