slight code tidy up