updated the tests