minor code fix up ..