Test mode tidy up