minor cleanup
minor code cleanup