minor code revision