Small tidyup.