modified the tests