test code tidy up