minor code cleanup
minor code cleanup