Tidy up tests
Tidy up tests