Minor code tidy