updated the tests a bit