minor code correction