Made one of the tests a little more flexible