Changed the tests slightly