minor code cleanup

minor code cleanup