small tidy up.