minor tidy up.