minor code bug fix