Very minor code tidy-up