fix a new bug
update the code for bug fix