minor code adjustment