Minor code tidy.