fix a bug
update the code