improving the tests