tidying up tests