small tidyup.