minor code follow up ..