tidying up the tests