minor code edit