minor tidyup