updated the tests a bit.