tests tidy up