Modified one of the tests