Minor code follow up .