fixed minor code