Tiny tidy up