Needed a change to one of the tests